Fonction torch.renorm de PyTorch

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

torch.renormC'est une fonction de PyTorch utilisée pour renormaliser les tenseurs. Elle normalise le tenseur selon la norme spécifiée, de sorte que la norme des éléments le long de la dimension spécifiée ne dépasse pas la valeur donnée.

Définition de fonction

torch.renorm(input, p, dim, maxnorm)

Description des paramètres :

  • input: tenseur d’entrée
  • p: Ordre de la norme
  • dim: Dimension pour la normalisation
  • maxnorm: Valeur maximale de la norme

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur
x = torch.tensor([[2.0, 4.0, 6.0], [3.0, 6.0, 9.0]])

# Effectuer une normalisation de norme L2 sur dim=1, avec une norme maximale de 1
y = torch.renorm(x, p=2, dim=1, maxnorm=1)
print(y)

Le résultat de sortie est:

tensor([[0.2673, 0.5345, 0.8018],
        [0.2673, 0.5345, 0.8018]])

Exemple

import torch

# Créer un tenseur
x = torch.tensor([[1.0, 2.0, 3.0]])

# Effectuer une normalisation de norme L1 sur dim=1
y = torch.renorm(x, p=1, dim=1, maxnorm=1)
print(y)

Le résultat de sortie est:

tensor([[0.1667, 0.3333, 0.5000]])

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

Autres extensions