PyTorch torch.mul Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.mulEs ist eine Funktion in PyTorch, die verwendet wird, um eine elementweise Multiplikation durchzuführen. Sie multipliziert zwei Tensoren oder einen Tensor und einen Skalar.

Dies ist eine der grundlegendsten mathematischen Operationen im Deep Learning.

Funktionsdefinition

torch.mul(input, other, out=None)

Parameters:

  • input(Tensor): Der erste Eingabe-Tensor.
  • other(Tensor oder float): Der zweite Eingabe-Tensor oder Skalar.
  • out(Tensor, optional): Der Ausgabe-Tensor.

Rückgabewert:

  • torch.Tensor: Gibt den multiplizierten Tensor zurück.

Anwendungsbeispiele

Beispiel 1: Elementweise Multiplikation von Tensoren

Beispiel

import torch

a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])

c = torch.mul(a, b)
print(c)

Das Ausgabeergebnis ist:

tensor([ 4, 10, 18])

Beispiel 2: Multiplikation mit einem Skalar

Beispiel

import torch

a = torch.tensor([1, 2, 3])

# Mit 2 multiplizieren
b = torch.mul(a, 2)
print(b)

# Du kannst auch den *-Operator verwenden
c = a * 2
print(c)

Das Ausgabeergebnis ist:

tensor([2, 4, 6])
tensor([2, 4, 6])

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen