detach(): 返回一个新的Tensor,但返回的结果是没有梯度的。
cpu():把gpu上的数据转到cpu上。
numpy():将tensor格式转为numpy。
def t2n(t):
return t.detach().cpu().numpy().astype(np.float)
该端代码表示将tensor类型的变量t转换为numpy浮点型
detach(): 返回一个新的Tensor,但返回的结果是没有梯度的。
cpu():把gpu上的数据转到cpu上。
numpy():将tensor格式转为numpy。
def t2n(t):
return t.detach().cpu().numpy().astype(np.float)
该端代码表示将tensor类型的变量t转换为numpy浮点型