PyTorch:torch.zeros_like()的用法

作用:产生一个与a相同shape的Tensor.

举例:

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


'''   运行结果   '''
a: tensor([[0.9285, 0.0966, 0.2180, 0.1234],
           [0.8930, 0.9143, 0.8272, 0.1228],
           [0.7441, 0.3232, 0.2867, 0.6598]])
b: tensor([[0., 0., 0., 0.],
           [0., 0., 0., 0.],
           [0., 0., 0., 0.]])

 

你可能感兴趣的:(Pytorch,深度学习)