代码拉取完成,页面将自动刷新
"""
tools to convert specified type
"""
import torch as t
import numpy as np
def tonumpy(data):
if isinstance(data, np.ndarray):
return data
if isinstance(data, t.Tensor):
return data.detach().cpu().numpy()
def totensor(data, cuda=True):
if isinstance(data, np.ndarray):
tensor = t.from_numpy(data)
if isinstance(data, t.Tensor):
tensor = data.detach()
if cuda:
tensor = tensor.cuda()
return tensor
def scalar(data):
if isinstance(data, np.ndarray):
return data.reshape(1)[0]
if isinstance(data, t.Tensor):
return data.item()
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。