elementAt method

Tensor<T> elementAt(
  1. int index, {
  2. int axis = 0,
})

Returns a view with the given axis resolved to index.

Implementation

Tensor<T> elementAt(int index, {int axis = 0}) => Tensor<T>.internal(
    type: type, layout: layout.elementAt(index, axis: axis), data: data);