[ 组合替代实现 ]torch.device¶

torch.device ¶

          torch.device(type, index)

         

Paddle 无此 API，需要组合实现。

          # PyTorch 写法
torch.device('cuda', 0)

# Paddle 写法
'gpu:0'

# PyTorch 写法
torch.device('cpu')

# Paddle 写法
'cpu'