记录 | python保存pytorch tensor至txt中的方法
  HRODlxSsrDy9 2023年12月19日 29 0


有时候,为了方便调试、定位问题,需要把 pytorch 中的 tensor 张量保存到 txt 中,方法如下,

比如需要保存的 pytorch 的 tensor 为 outputs,其维度为 [1, 1, 250, 11],则:

with open("outputs.txt", "w") as file:
     for i in range(250):
         for j in range(11):
             file.write(str(outputs[0][0][i][j].item()))
             file.write('\n')

这样就把张量 outputs 的值保存到了 outputs.txt 中了。


【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年12月19日 0

暂无评论

推荐阅读
HRODlxSsrDy9