unsqueeze

fun unsqueeze(a: Tensor<*, *>, dim: Int): Tensor<*, *>(source)

Add a dimension of size 1 at the given position.