PyTorch torch.ones-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.onesist eine Funktion in PyTorch, die zum Erstellen eines Tensors aus lauter Einsen verwendet wird. Sie erstellt einen Tensor mit der angegebenen Form, bei dem alle Elemente auf 1 initialisiert sind.

Diese wird im Deep Learning häufig verwendet, um bestimmte Parameter zu initialisieren, Masken zu erstellen oder als Basiswerte für mathematische Operationen zu dienen.

Funktionsdefinition

torch.ones(*size, dtype=None, device=None, requires_grad=False, pin_memory=False)

Parameters:

  • *size(int): Die Form des Tensors, z. B.3、(3, 4)、(2, 3, 4)usw.
  • dtype(torch.dtype, optional): Gibt den Datentyp des Tensors an, Standard isttorch.float32。
  • device(torch.device, optional): Gibt das Gerät an, auf dem der Tensor gespeichert wird.
  • requires_grad(bool, optional): Gibt an, ob Gradienten berechnet werden sollen.
  • pin_memory(bool, optional): Ob Pinned Memory verwendet werden soll.

Rückgabewert:

  • torch.Tensor: Gibt einen Tensor aus lauter Einsen zurück.

Anwendungsbeispiele

Beispiel 1: 1D-Einsen-Tensor erstellen

Beispiel

import torch

# Einen Tensor aus lauter Einsen mit 5 Elementen erstellen
x = torch.ones(5)

print(x)

Das Ausgabeergebnis ist:

tensor([1., 1., 1., 1., 1.])

Beispiel 2: 2D-Einsen-Tensor erstellen

Beispiel

import torch

# Einen 3x4-Tensor aus lauter Einsen (Matrix) erstellen
x = torch.ones(3, 4)

print(x)
print(x.shape)

Das Ausgabeergebnis ist:

tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
torch.Size([3, 4])

Beispiel 3: Einen 3D-Einser-Tensor erstellen

Beispiel

import torch

# Einen 2x3x4-Einser-Tensor erstellen
x = torch.ones(2, 3, 4)

print(x.shape)

Das Ausgabeergebnis ist:

torch.Size([2, 3, 4])

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen