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

该 api 有两组参数列表重载，因此有两组差异分析。

torch.device ¶

          torch.device(type, index)

         

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

转写示例¶

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

# Paddle 写法
'gpu:0'

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

# Paddle 写法
'cpu'

         

torch.device ¶

          torch.device(device)

         

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

转写示例¶

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

# Paddle 写法
'cpu'

# PyTorch 写法
torch.device(1)

# Paddle 写法
"gpu:1"

# PyTorch 写法
torch.device("cuda:0")

# Paddle 写法
"gpu:0"

         