square

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

Creates a square convolution kernel

outputChannels

number of output channels

inputChannels

number of input channels

kernelSize

size of the square kernel (both height and width)