PyTorch torch.prod Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.prodist eine Funktion in PyTorch, die verwendet wird, um das Produkt aller Elemente in einem Tensor zurückzugeben.

Funktionsdefinition

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

Anwendungsbeispiel

Beispiel

import torch

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

# Gibt das Produkt aller Elemente zurück
print("Produkt aller Elemente:", torch.prod(x))

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

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen