Squeeze

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

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