File size: 434 Bytes
88b0dcb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
"""
@date: 2021/8/4
@description:
"""
import numpy as np
import torch
def tensor2np(t: torch.Tensor) -> np.array:
if isinstance(t, torch.Tensor):
if t.device == 'cpu':
return t.detach().numpy()
else:
return t.detach().cpu().numpy()
else:
return t
def tensor2np_d(d: dict) -> dict:
output = {}
for k in d.keys():
output[k] = tensor2np(d[k])
return output
|