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

unsqueeze

fun <I, T : DType, V> Transform<I, Tensor<T, V>>.unsqueeze(ctx: ExecutionContext, dim: Int): Transform<I, Tensor<T, V>>(source)

Chains an Unsqueeze transform that adds a dimension.

Parameters

ctx

The execution context

dim

The position to add the new dimension