mindspore.ops.Unstack
- class mindspore.ops.Unstack(axis=0)[源代码]
根据指定轴对输入矩阵进行分解。
更多参考详见
mindspore.ops.unstack()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> 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]))