skainet-data-transform/sk.ainet.data.transform/Squeeze

Squeeze

class Squeeze<T : DType, V>(ctx: ExecutionContext, val dim: Int? = null) : TensorTransform<T, V> (source)

Removes a dimension of size 1 at the specified position.

Parameters

T

The tensor data type

V

The value type

ctx

The execution context for tensor operations

dim

The dimension to remove (must have size 1), or null to remove all size-1 dims

Constructors

constructor(ctx: ExecutionContext, dim: Int? = null)

Properties

val dim: Int?

Functions

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