对于大型数组,如果不进行原地操作则可能造成内存溢出的问题。在使用时需要注意。
示例如下:
import torch
x = torch.tensor([1,2])
y = torch.tensor([1,2])
print('id(y):',id(y))
y = y + x
print('id(y):',id(y))
print(y)
id(y): 2290817994240
id(y): 2290715213072
tensor([2, 4])
print('id(y):',id(y))
y += x
print('id(y):',id(y))
print(y)
id(y): 2290715213072
id(y): 2290715213072
tensor([3, 6])
关键区别: y = y + x 和 y += x