Conv2d

class Conv2d<T : DType, V> @JvmOverloads constructor(val inChannels: Int, val outChannels: Int, val kernelSize: Pair<Int, Int>, val stride: Pair<Int, Int> = 1 to 1, val padding: Pair<Int, Int> = 0 to 0, val dilation: Pair<Int, Int> = 1 to 1, val groups: Int = 1, val bias: Boolean = true, val name: String = "Conv2d", initWeights: Tensor<T, V>, initBias: Tensor<T, V>? = null, val trainable: Boolean = true) : Module<T, V> , ModuleParameters<T, V> (source)

2D Convolutional layer that applies a convolution operation over 2D input.

inChannels

Number of input channels

outChannels

Number of output channels/filters

kernelSize

Size of the convolving kernel (height, width)

stride

Stride of the convolution (default: 1, 1)

padding

Padding added to all sides of the input (default: 0, 0)

dilation

Spacing between kernel elements (default: 1, 1)

groups

Number of blocked connections from input channels to output channels (default: 1)

bias

Whether to add a learnable bias to the output (default: true)

name

Name of the module

initWeights

Initial weights tensor

initBias

Initial bias tensor (if bias is true)

Constructors

constructor(inChannels: Int, outChannels: Int, kernelSize: Pair<Int, Int>, stride: Pair<Int, Int> = 1 to 1, padding: Pair<Int, Int> = 0 to 0, dilation: Pair<Int, Int> = 1 to 1, groups: Int = 1, bias: Boolean = true, name: String = "Conv2d", initWeights: Tensor<T, V>, initBias: Tensor<T, V>? = null, trainable: Boolean = true)