.detach().cpu().numpy()该段代码的作用
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()
为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。
更多推荐
10
0- 0
已为社区贡献1条内容
所有评论(0)