花费 22 ms
2.8 将tensor移动到GPU上

在Pytorch中,所有对tensor的操作,都是由GPU-specific routines完成的。tensor的device属性来控制tensor在计算机中存放的位置。 我们可以在tensor的 ...

Thu Jul 30 22:27:00 CST 2020 0 1693
2.7 序列化(保存)tensor

可以直接调用tensor的save和load方法对tensor进行保存和读取. 保存: 读取: 或者 但是这样保存下来的只能被Pytorch所识别。可以利用 ...

Thu Jul 30 19:57:00 CST 2020 0 809

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM