torch.Tensor.bernoulli_(p=0.5, *, generator=None)
Paddle 无此 API,需要组合实现。
# Pytorch 写法
input.bernoulli_(p=x)
# Paddle 写法
paddle.assign(paddle.bernoulli(paddle.broadcast_to(x, input.shape)), input)
# Pytorch 写法
input.bernoulli_(p=x)
# Paddle 写法
tensor = paddle.to_tensor([x])
paddle.assign(paddle.bernoulli(paddle.broadcast_to(tensor, input.shape)), input)