Upsample2d

class Upsample2d<T : DType, V>(val scale: Pair<Int, Int> = 2 to 2, val mode: UpsampleMode = UpsampleMode.Nearest, val alignCorners: Boolean = false, val name: String = "Upsample2d") : Module<T, V> (source)

2D upsampling layer for increasing spatial resolution.

Supports nearest-neighbor mode (typical for YOLO necks). Bilinear can be wired once backends add support.

Constructors

constructor(scale: Pair<Int, Int> = 2 to 2, mode: UpsampleMode = UpsampleMode.Nearest, alignCorners: Boolean = false, name: String = "Upsample2d")

open override val modules: List<Module<T, V>>

open override val name: String

Human-readable name of the node, typically the module's name.

open override fun forward(input: Tensor<T, V>, ctx: ExecutionContext): Tensor<T, V>