detach(): 返回一个新的Tensor,但返回的结果是没有梯度的。
cpu():把gpu上的数据转到cpu上。
numpy():将tensor格式转为numpy。
如图所示:
out = logits.detach().cpu().numpy()
detach(): 返回一个新的Tensor,但返回的结果是没有梯度的。
cpu():把gpu上的数据转到cpu上。
numpy():将tensor格式转为numpy。
如图所示:
out = logits.detach().cpu().numpy()