[ 仅 paddle 参数更多 ] torch.cuda.HalfTensor¶

torch.cuda.HalfTensor ¶

          torch.cuda.HalfTensor(data)

         

          paddle.to_tensor(data, dtype='float16', place='gpu')

         

Paddle 比 PyTorch 支持更多参数，具体如下：

PyTorch	PaddlePaddle	备注
-	dtype	Tensor 的数据类型，PyTorch 无此参数，Paddle 需设置为 'float16'。
-	place	Tensor 的设备，PyTorch 无此参数，Paddle 需设置为 'gpu' 。