[ 仅 paddle 参数更多 ] torch.cuda.ByteTensor¶
torch.cuda.ByteTensor¶
torch.cuda.ByteTensor(data)
paddle.to_tensor¶
paddle.to_tensor(data, dtype='uint8', place='gpu', stop_gradient=True)
Paddle 比 PyTorch 支持更多参数,具体如下:
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
data | data | 要转换的数据。 |
- | dtype | Tensor 的数据类型,PyTorch 无此参数,Paddle 需设置为 'uint8'。 |
- | place | Tensor 的设备,PyTorch 无此参数,Paddle 需设置为 'gpu' 。 |
- | stop_gradient | 是否阻断 Autograd 的梯度传导。PyTorch 无此参数,Paddle 保持默认即可。 |