towhee.hub.builtin.operators.tensor_like.tensor_split¶
- class towhee.hub.builtin.operators.tensor_like.tensor_split(axis=0)[source]¶
Bases:
object
Split the input array along a axis.
- Parameters:
axis (int, optional) – the axis along which to split the array. Defaults to 0.
- Returns:
list of result arrays.
- Return type:
[ndarray]
Examples:
>>> import numpy as np >>> from towhee.functional import DataCollection >>> dc = ( ... DataCollection.range(3) ... .map(lambda x: np.array([[x, x*2]])) ... .tensor_split(axis=1) ... ) >>> dc.to_list() [[array([[0]]), array([[0]])], [array([[1]]), array([[2]])], [array([[2]]), array([[4]])]]
Methods
Attributes
metainfo
- __call__(*args, **kws)¶
Call self as a function.