MindSpore中张量和Numpy的互相转换

NumPy 是 Python 语言的一个扩展程序库,支持大量的维度数组与矩阵运算,此外也针对数组运算提供大量的数学函数库。所以下面展示下MindSpore中张量和Numpy类型的互相转换。

1. 张量转换为NumPy

   zeros = ops.Zeros()

   output = zeros((2,2), mstype.float32)

   print("output: {}".format(type(output)))

   n_output = output.asnumpy()

   print("n_output: {}".format(type(n_output)))

   输出：

   output: <class 'mindspore.common.tensor.Tensor'> 

   n_output: <class 'numpy.ndarray'>

2. NumPy转换为张量

   output = np.array([1, 0, 1, 0])

   print("output: {}".format(type(output)))

   t_output = Tensor(output)

   print("t_output: {}".format(type(t_output)))

   输出：

   output: <class 'numpy.ndarray'> 

   t_output: <class 'mindspore.common.tensor.Tensor'>