Tensor数组和numpy数组在使用上非常相似，我正在写一个函数，它可以求除其中的一个，但是我需要找到一种方法，用同一个调用同时得到Tensor数组和np数组的形状，这可能吗？
iezvtpos1#
我发现您可以使用'.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])
两者都可以按元素访问
1条答案
按热度按时间iezvtpos1#
我发现您可以使用'.shape'表示以下两种情况：
印刷品
两者都可以按元素访问