Returns a view with a single-element axis at axis removed.
axis
Tensor<T> collapse({int axis = 0}) => Tensor<T>.internal( type: type, layout: layout.collapse(axis: axis), data: data);