towhee.hub.builtin.operators.tensor_like.tensor_unstack

class towhee.hub.builtin.operators.tensor_like.tensor_unstack(axis=0)

Bases: object

Unstack an array along given axis.

Parameters:

axis (int, optional) – the axis along which to unstack the array. Defaults to 0.

Returns:

sequence 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_unstack(axis=0)
... )
>>> 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.

__init__(axis=0)