.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)