Loading [Contrib]/a11y/accessibility-menu.js
\u200E
文档[ torch 参数更多 ]torch.bernoulli Edit on Github
[ torch 参数更多 ]torch.bernoulli
torch.bernoulli(input,
*,
generator=None,
out=None)
paddle.bernoulli(x,
name=None)
PyTorch 相比 Paddle 支持更多其他参数,具体如下:
参数映射
PyTorch |
PaddlePaddle |
备注 |
input |
x |
表示输入的 Tensor,仅参数名不一致。 |
generator |
- |
用于采样的伪随机数生成器, Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。 |
out |
- |
表示输出的 Tensor,Paddle 无此参数,需要转写。 |
转写示例
out:指定输出
torch.bernoulli(torch.tensor([3, 5]), out=y)
paddle.assign(paddle.bernoulli(paddle.to_tensor([3, 5])), y)
文档反馈