[Pytorch] torch.zeros contre torch.zeros_like

Je suis curieux de connaître la différence entre torch.zeros et torch.zeros_like, alors je vais l'écrire.

Comment utiliser torch.zeros

Renvoie un tenseur avec une valeur de 0.

>>> torch.zeros(3, 2)
tensor([[0., 0.],
        [0., 0.],
        [0., 0.]])
>>> torch.zeros(3)
tensor([0., 0., 0.])

Comment utiliser torch.zeros_like

Il est utilisé lorsque vous souhaitez définir toutes les valeurs d'un certain tenseur à 0.

>>> input = torch.empty(3, 2)
tensor([[3.2561e+09, 3.0936e-41],
        [0.0000e+00, 0.0000e+00],
        [3.2557e+09, 3.0936e-41]])
>>> torch.zeros_like(input)
tensor([[0., 0.],
        [0., 0.],
        [0., 0.]])

Dans l'exemple ci-dessus, toutes les valeurs d'entrée sont définies sur 0.