chw

fun chw(channels: Int, height: Int, width: Int): ShapeBuilder<T, V>(source)

Creates a single image tensor in CHW format

Parameters

channels

number of channels (e.g., 3 for RGB, 1 for grayscale)

height

image height in pixels

width

image width in pixels