2020
10-08
10-08
浅谈PyTorch中in-place operation的含义
in-placeoperation在pytorch中是指改变一个tensor的值的时候,不经过复制操作,而是直接在原来的内存上改变它的值。可以把它成为原地操作符。在pytorch中经常加后缀“_”来代表原地in-placeoperation,比如说.add_()或者.scatter()。python里面的+=,*=也是in-placeoperation。下面是正常的加操作,执行结束加操作之后x的值没有发生变化:importtorchx=torch.rand(2)#tensor([0.8284,0.5539])print(x)y=torch.rand(2)print(x...
继续阅读 >