Indexing Routines

Generating index arrays

mars.tensor.nonzero Return the indices of the elements that are non-zero.
mars.tensor.where Return elements, either from x or y, depending on condition.
mars.tensor.indices Return a tensor representing the indices of a grid.
mars.tensor.ogrid Construct a multi-dimensional “meshgrid”.
mars.tensor.unravel_index Converts a flat index or tensor of flat indices into a tuple of coordinate tensors.

Indexing-like operations

mars.tensor.take Take elements from a tensor along an axis.
mars.tensor.choose Construct a tensor from an index tensor and a set of tensors to choose from.
mars.tensor.compress Return selected slices of a tensor along given axis.
mars.tensor.diag Extract a diagonal or construct a diagonal tensor.