DepthwiseSeparableConv2d
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, bias: Boolean = true, name: String = "DepthwiseSeparableConv2d", initDepthwiseWeights: Tensor<T, V>, initPointwiseWeights: Tensor<T, V>, initDepthwiseBias: Tensor<T, V>? = null, initPointwiseBias: Tensor<T, V>? = null)(source)
Parameters
inChannels
Number of input channels
outChannels
Number of output channels/filters
kernelSize
Size of the convolving kernel for depthwise convolution (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)
bias
Whether to add learnable bias to both depthwise and pointwise layers (default: true)
name
Name of the module
initDepthwiseWeights
Initial weights for depthwise convolution
initPointwiseWeights
Initial weights for pointwise convolution
initDepthwiseBias
Initial bias for depthwise convolution (if bias is true)
initPointwiseBias
Initial bias for pointwise convolution (if bias is true)