Fonction torch.mul de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.mulIl s'agit d'une fonction de PyTorch utilisée pour effectuer une multiplication élément par élément. Elle multiplie deux tenseurs ou un tenseur et un scalaire.

Il s'agit de l'une des opérations mathématiques les plus fondamentales en apprentissage profond.

Définition de fonction

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

Parameters:

  • input(Tensor) : Le premier tenseur d'entrée.
  • other(Tensor or float) : Le deuxième tenseur d'entrée ou un scalaire.
  • out(Tensor, optionnel) : Le tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie le tenseur multiplié.

Exemples d'utilisation

Exemple 1 : Multiplication élément par élément de tenseurs

Exemple

import torch

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

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

Le résultat de sortie est:

tensor([ 4, 10, 18])

Exemple 2 : Multiplication par un scalaire

Exemple

import torch

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

# Multiplier par 2
b = torch.mul(a, 2)
print(b)

# Vous pouvez également utiliser l'opérateur *
c = a * 2
print(c)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions