Fonction torch.prod de PyTorch

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

torch.prodest une fonction dans PyTorch utilisée pour renvoyer le produit de tous les éléments d'un tenseur.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

# Renvoyer le produit de tous les éléments
print("Produit de tous les éléments :", torch.prod(x))

# Produit selon dim=0
y = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("produit dim=0 :", torch.prod(y, dim=0))
print("produit dim=1 :", torch.prod(y, dim=1))

Le résultat de sortie est:

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

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

Autres extensions