[PYTHON] [Pytorch] torch.zeros vs torch.zeros_like

Ich bin neugierig auf den Unterschied zwischen torch.zeros und torch.zeros_like, also schreibe ich es.

Wie benutzt man torch.zeros?

Gibt einen Tensor mit dem Wert 0 zurück.

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

Verwendung von torch.zeros_like

Es wird verwendet, wenn Sie alle Werte eines bestimmten Tensors auf 0 setzen möchten.

>>> 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.]])

Im obigen Beispiel werden alle Eingabewerte auf 0 gesetzt.