\u200E

[ torch 参数更多 ] torch.Tensor.bfloat16¶

torch.Tensor.bfloat16 ¶

          torch.Tensor.bfloat16(memory_format=torch.preserve_format)

         

paddle.Tensor.astype ¶

          paddle.Tensor.astype('bfloat16')

         

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

参数映射¶

PyTorch	PaddlePaddle	备注
memory_format	-	表示内存格式， Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。

转写示例¶

          # torch 写法
x = torch.randn(3, 3)
y = x.bfloat16()

# paddle 写法
x = paddle.randn([3, 3])
y = x.astype('bfloat16')

         