PyTorch:torch.zeros_like()的用法

举例：

import torch
b = torch.rand(5,6) # 产生一个5行6列的0～1的随机Tensor
c = torch.zeros_like(b) # 产生一个与b相同shape的全零Tensor
print('b:',b)
print('c:',c)

作用：产生一个与b相同shape的Tensor

作用
执行原地操作

before = id(Y)
Y = Y + X
id(Y) == before

Z = torch.zeros_like(Y)
print('id(Z):', id(Z))
Z[:] = X + Y
print('id(Z):', id(Z))