ImageTensor

Creates image tensors in BCHW (Batch, Channel, Height, Width) format

Constructors

constructor(builder: TensorBuilder<T, V>)

fun bchw(batchSize: Int, channels: Int, height: Int, width: Int): ShapeBuilder<T, V>

Creates a 4D tensor for image batch in BCHW format

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

Creates a single image tensor in CHW format

Creates a grayscale image tensor (single channel)

fun rgb(height: Int, width: Int): ShapeBuilder<T, V>

Creates an RGB image tensor (3 channels)

fun rgba(height: Int, width: Int): ShapeBuilder<T, V>

Creates an RGBA image tensor (4 channels)