和python差不多,直接上id代码了
索引、view不会开辟新内存
而y=y+x会开辟,如果不要开辟,用y[:]=y+x或torch.add(x,y,out=y)
x=torch.tensor([1,2])
y=torch.tensor([1,2])
id_before=id(y)
y=y+x
print(id(y)==id_before)
x=torch.tensor([1,2])
y=torch.tensor([1,2])
id_before=id(y)
y[:]=y+x
print(id(y)==id_before)
'''
False
True
'''
x=torch.tensor([1,2])
y=torch.tensor([3,4])
id_before=id(y)
torch.add(x,y,out=y)
print(id(y)==id_before)
'''
True
'''

被折叠的 条评论
为什么被折叠?



