[ 仅 paddle 参数更多 ] torch.cuda.FloatTensor

torch.cuda.FloatTensor

torch.cuda.FloatTensor(data)

paddle.to_tensor

paddle.to_tensor(data, dtype='float32', place='gpu', stop_gradient=True)

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

参数映射

PyTorch PaddlePaddle 备注
data data 要转换的数据。
- dtype Tensor 的数据类型,PyTorch 无此参数,Paddle 需设置为 'float32'。
- place Tensor 的设备,PyTorch 无此参数,Paddle 需设置为 'gpu' 。
- stop_gradient 是否阻断 Autograd 的梯度传导。PyTorch 无此参数,Paddle 保持默认即可。