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

unsqueeze

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

Chains an Unsqueeze transform.