获取张量和 np 数组的形状

【问题标题】：Get shape of both tensor and np array获取张量和 np 数组的形状
【发布时间】：2023-01-13 04:35:50
【问题描述】：

Tensor 和 numpy 数组在使用上非常相似。我正在编写一个函数，但我需要找到一种方法来通过相同的调用获取张量和 np 数组的形状。那可能吗？

【问题讨论】：

【解决方案1】：

我发现您可以对两者使用“.shape”：

myNpArray = np.zeros((1,10,256))
print(myNpArray.shape)

myTensor = torch.zeros((1, 10, 256))
print(myTensor.shape)

印刷

(1, 10, 256)
torch.Size([1, 10, 256])

两者都可以按元素访问

【讨论】：