Función torch.prod de PyTorch

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

torch.prodEs una función en PyTorch que se utiliza para devolver el producto de todos los elementos de un tensor.

Definición de la función

torch.prod(input, dim, keepdim=False)

Ejemplo de uso

Ejemplo

import torch

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

# Devuelve el producto de todos los elementos
print("Producto de todos los elementos:", torch.prod(x))

# Producto a lo largo de dim=0
y = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("Producto dim=0:", torch.prod(y, dim=0))
print("Producto dim=1:", torch.prod(y, dim=1))

El resultado de salida es:

所有元素乘积: tensor(24)
dim=0 乘积: tensor([4, 10, 18])
dim=1 乘积: tensor([  6, 120])

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

Otras extensiones