NumPy 是 Python 语言的一个扩展程序库,支持大量的维度数组与矩阵运算,此外也针对数组运算提供大量的数学函数库。所以下面展示下MindSpore中张量和Numpy类型的互相转换。
-
张量转换为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'>
-
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'>
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)