Función torch.mul de PyTorch

Pytorch torch 参考手册Manual de referencia de torch de PyTorch

torch.mulEs una función en PyTorch que se utiliza para realizar multiplicación elemento a elemento. Multiplica dos tensores o un tensor y un escalar.

Esta es una de las operaciones matemáticas más básicas en el aprendizaje profundo.

Definición de la función

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

Parámetros:

  • input(Tensor): El primer tensor de entrada.
  • other(Tensor o float): El segundo tensor de entrada o escalar.
  • out(Tensor, opcional): El tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve el tensor resultante de la multiplicación.

Ejemplo de uso

Ejemplo 1: Multiplicación elemento a elemento de tensores

Ejemplo

import torch

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

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

El resultado de salida es:

tensor([ 4, 10, 18])

Ejemplo 2: Multiplicar por un escalar

Ejemplo

import torch

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

# Multiplicar por 2
b = torch.mul(a, 2)
print(b)

# También se puede usar el operador *
c = a * 2
print(c)

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de torch de PyTorch

Otras extensiones