GroupedConv2d

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

Parameters

inChannels

Number of input channels

outChannels

Number of output channels/filters

kernelSize

Size of the convolving kernel (height, width)

groups

Number of groups to divide channels into

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)

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)