Reshape

Reshapes a tensor to a new shape without changing the data.

The total number of elements must remain the same.

Usage

val reshape = Reshape<FP32, Float>(ctx, Shape(1, 3, 224, 224))
val batched = reshape.apply(imageTensor)  // Add batch dimension

The tensor data type

The value type

ctx

The execution context for tensor operations

newShape

The target shape

constructor(ctx: ExecutionContext, newShape: Shape)

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