[ 仅 paddle 参数更多 ]torch.nn.Module.bfloat16

torch.nn.Module.bfloat16

torch.nn.Module.bfloat16()

paddle.nn.Layer.to

paddle.nn.Layer.to(dtype=paddle.bfloat16)

Paddle 相比 PyTorch 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
- dtype 转换的数据类型,Paddle 为 paddle.bfloat16,需要转写。

转写示例

dtype 参数:转换的数据类型

# PyTorch 写法:
module = torch.nn.Module()
module.bfloat16()

# Paddle 写法:
module = paddle.nn.Layer()
module.to(dtype=paddle.bfloat16)