比较与torch.Tensor.size的差异
torch.Tensor.size
torch.Tensor.size() -> Tensor
更多内容详见torch.Tensor.size。
mindspore.Tensor.shape
mindspore.Tensor.shape
更多内容详见mindspore.Tensor.shape。
差异对比
PyTorch:size()方法,返回Tensor的shape。
MindSpore:功能一致,但是mindspore.Tensor.shape为属性,不为方法。
代码示例
# PyTorch
import torch
input = torch.randn(3, 4, 5)
print(input.size())
# torch.Size([3, 4, 5])
# MindSpore
import mindspore.ops as ops
input = ops.randn(3, 4, 5)
print(input.shape)
# (3, 4, 5)