Translates slice objects to concatenation along the second axis.
Translates slice objects to concatenation along the first axis.
Return the indices of the elements that are non-zero.
Return elements, either from x or y, depending on condition.
Return a tensor representing the indices of a grid.
Construct a multi-dimensional “meshgrid”.
Converts a flat index or tensor of flat indices into a tuple of coordinate tensors.
Take elements from a tensor along an axis.
Construct a tensor from an index tensor and a set of tensors to choose from.
Return selected slices of a tensor along given axis.
Extract a diagonal or construct a diagonal tensor.