[ torch 参数更多 ] torch.Tensor.bool¶
torch.Tensor.bool¶
torch.Tensor.bool(memory_format=torch.preserve_format)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
memory_format | - | 表示内存格式, Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。 |
转写示例¶
# torch 写法
x = torch.tensor([-1, 0, 1])
y = x.bool()
# paddle 写法
x = paddle.to_tensor([-1, 0, 1])
y = x.astype('bool')