Conv2d
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)(source)
Parameters
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)