ConvKernel

Creates convolution kernel tensors

Constructors

constructor(builder: TensorBuilder<T, V>)

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

Creates a 3x3 convolution kernel (most common)

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

Creates a 5x5 convolution kernel

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

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

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

Creates a 1x1 convolution kernel (pointwise convolution)

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

Creates a square convolution kernel