skainet-lang-core/sk.ainet.lang.tensor.dsl/ConvKernel/pointwise

pointwise

fun pointwise(outputChannels: Int, inputChannels: Int): ShapeBuilder<T, V>(source)

Creates a 1x1 convolution kernel (pointwise convolution)

Parameters

outputChannels

number of output channels

inputChannels

number of input channels