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