Html/CSS
存txt 标签描述

有时候,为了方便调试、定位问题,需要把pytorch中的tensor张量保存到txt中,方法如下, 比如需要保存的pytorch的tensor为outputs,其维度为[1,1,250,11],则: withopen("outputs.txt","w")asfile: foriinrange(250): forjinrange(11): file.write(str(outputs[0][0][i][j].item())) file.write('\n') 这样就把张量outputs的值保存到了outputs.txt中了。