Skip to main content

//clika-runtime/io.clika.runtime/Ops/conv2d

conv2d

[common]
fun conv2d(input: Tensor, weight: Tensor, bias: Tensor? = null, stride: LongArray = longArrayOf(1, 1), padding: LongArray = longArrayOf(0, 0), dilation: LongArray = longArrayOf(1, 1), groups: Long = 1, activation: Activation = Activation.IDENTITY): Tensor

conv2d(input: Tensor, weight: Tensor, bias: Tensor? = null, stride: LongArray = longArrayOf(1, 1), padding: LongArray = longArrayOf(0, 0), dilation: LongArray = longArrayOf(1, 1), groups: Long = 1L, activation: Activation = Activation.IDENTITY): the conv2d operator. 2-D convolution over a channels-last input. Layout is channels-last: input [N, spatial.., C], weight OHWI [O, K.., C/groups] (output channels first, kernel dims, then the per-group input channels), optional bias [O].