[ 仅 paddle 参数更多 ] torch.cuda.ShortTensor¶
torch.cuda.ShortTensor¶
torch.cuda.ShortTensor(data)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
- | dtype | Tensor 的数据类型,PyTorch 无此参数,Paddle 需设置为 'int16'。 |
- | place | Tensor 的设备,PyTorch 无此参数,Paddle 需设置为 'gpu' 。 |