Returns a view with the given axis resolved to index.
axis
index
Tensor<T> elementAt(int index, {int axis = 0}) => Tensor<T>.internal( type: type, layout: layout.elementAt(index, axis: axis), data: data);