比较与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)