Multi-Output

split

def split(x: 'Tensor', num_splits: 'int', axis: 'int' = 0) -> 'list':

Split a tensor into multiple equal chunks along an axis.

---

chunk

def chunk(x: 'Tensor', chunks: 'int', axis: 'int' = 0) -> 'list':

Split a tensor into a specified number of chunks.

---

unbind

def unbind(x: 'Tensor', axis: 'int' = 0) -> 'list':

Remove a dimension and return list of slices.

---

minmax

def minmax(x: 'Tensor') -> 'dict[str, Tensor]':

Return both min and max of a tensor as a dict with ‘min’ and ‘max’ keys.

---