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