比较与torch.randint的差异

查看源文件

torch.randint

torch.randint(low=0, high, size, *, generator=None, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)

更多内容详见torch.randint

mindspore.ops.randint

mindspore.ops.randint(low, high, size, seed=None, *, dtype=None)

更多内容详见mindspore.ops.randint

差异对比

PyTorch:low 为可选输入,默认值为0。

MindSpore:low 为必选输入,无默认值。

分类

子类

PyTorch

MindSpore

差异

参数

参数1

low

low

PyTorch中 low 有默认值0,MindSpore不存在默认值

参数2

high

high

无差异

参数3

size

size

无差异

参数4

generator

-

通用差异

参数5

out

-

通用差异

参数6

dtype

dtype

无差异

参数7

layout

-

通用差异

参数8

device

-

通用差异

参数9

requires_grad

-

通用差异

参数10

-

seed

通用差异

代码示例

# PyTorch
import torch

# PyTorch 无需传入low的值,相当于MindSpore中low=0。
x = torch.randint(10, (3, 3))
print(tuple(x.shape))
# (3, 3)

# MindSpore
import mindspore

# MindSpore 必须将torch中low的默认值(此处为0),作为输入传入。
x = mindspore.ops.randint(0, 10, (3, 3))
print(x.shape)
# (3, 3)