как libtorch получает тип тензорного устройства? - PullRequest
0 голосов
/ 03 марта 2020

Есть ли у libtorch метод получения тензорного или модельного типа устройства.

Так же, как xxx.device в pytorch?

import torch
tensor = torch.rand(3,4)
print(tensor.device)
...