numpy array可以生成一个矩阵,那么怎么用numpy生成一个三维的tensor?
1个回答
numpy.array可以用来构造任意维度的tensor,而不是局限于矩阵,完全取决于你用了多少括号
>> import numpy as np
>> m = np.array([[[1, 2, 3], [1, 2, 3]], [[1, 2, 5], [1, 2, 5]]])
>> m.shape
(2, 2, 3)
>> m
array([[[1, 2, 3],
[1, 2, 3]],
[[1, 2, 5],
[1, 2, 5]]])