mars.tensor.c_
Translates slice objects to concatenation along the second axis.
mars.tensor.r_
Translates slice objects to concatenation along the first axis.
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.
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.