Unsqueeze

Adds a dimension of size 1 at the specified position.

Usage

val unsqueeze = Unsqueeze<FP32, Float>(ctx, dim = 0)
val batched = unsqueeze.apply(imageTensor)  // [C, H, W] -> [1, C, H, W]

The tensor data type

The value type

ctx

The execution context for tensor operations

dim

The dimension to add

constructor(ctx: ExecutionContext, dim: Int)

open override fun apply(input: Tensor<T, V>): Tensor<T, V>

Applies this transformation to the given input.

open override fun getOutputShape(inputShape: Shape): Shape

By default, tensor transforms preserve the input shape. Override this method if the transform changes the shape.

open override fun toString(): String