[torch 参数更多 ]torch.randint¶

torch.randint¶

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

paddle.randint¶

paddle.randint(low=0,
               high=None,
               shape=[1],
               dtype=None,
               name=None)

Pytorch 相比 Paddle 支持更多其他参数，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | low | low | 表示生成的随机值范围的下限(区间一般包含)。 | | high | high | 表示生成的随机值范围的上限(区间一般不包含)。 | | size | shape | 表示输出形状大小。 | | generator | - | 用于采样的伪随机数生成器，Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。 | | out | - | 表示输出的 Tensor ， Paddle 无此参数，需要转写。 | | dtype | dtype | 表示数据类型。 | | layout | - | 表示布局方式， Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。 | | device | - | 表示 Tensor 存放设备位置，Paddle 无此参数，需要转写。 | | requires_grad | - | 表示是否计算梯度， Paddle 无此参数，需要转写。 |

转写示例¶

# Pytorch 写法
torch.randint(10, (2, 2), out=y)

# Paddle 写法
paddle.assign(paddle.randint(10, shape=[2, 2]), y)

# Pytorch 写法
x = torch.randint(10, (2, 2), requires_grad=True)

# Paddle 写法
x = paddle.randint(10, shape=[2, 2])
x.stop_gradient = False

# Pytorch 写法
torch.randint(10, (2, 2), device=torch.device('cpu'))

# Paddle 写法
y = paddle.randint(10, shape=[2, 2])
y.cpu()