Fonction torch.hypot de PyTorch

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

torch.hypotC'est une fonction dans PyTorch utilisée pour calculer la longueur de l'hypoténuse d'un triangle rectangle, c'est-à-dire sqrt(a^2 + b^2).

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

# Créer deux côtés de l'angle droit
a = torch.tensor([3.0, 5.0])
b = torch.tensor([4.0, 12.0])

# Calculer la longueur de l'hypoténuse
c = torch.hypot(a, b)
print("Hypoténuse :", c)

Le résultat de sortie est:

斜边: tensor([ 5., 13.])

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

Autres extensions