mindspore.ops.Unstack

class mindspore.ops.Unstack(axis=0)[source]

Unstacks tensor in specified axis.

Refer to mindspore.ops.unstack() for more detail.

Supported Platforms:: Ascend GPU CPU

Examples

>>> unstack = ops.Unstack()
>>> input_x = Tensor(np.array([[1, 1, 1, 1], [2, 2, 2, 2]]))
>>> output = unstack(input_x)
>>> print(output)
(Tensor(shape=[4], dtype=Int64, value= [1, 1, 1, 1]), Tensor(shape=[4], dtype=Int64, value= [2, 2, 2, 2]))