oihw

fun oihw(outputChannels: Int, inputChannels: Int, kernelHeight: Int, kernelWidth: Int): ShapeBuilder<T, V>(source)

Creates a 4D convolution kernel in OIHW format (Output, Input, Height, Width)

Parameters

outputChannels

number of output channels

inputChannels

number of input channels

kernelHeight

kernel height

kernelWidth

kernel width