Función torch.renorm de PyTorch

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

torch.renormEs una función de PyTorch utilizada para renormalizar tensores. Normaliza el tensor según la norma especificada, de modo que la norma de los elementos en la dimensión especificada no supere el valor dado.

Definición de la función

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

Descripción de parámetros:

  • input: tensor de entrada
  • p: orden de la norma
  • dim: dimensión de normalización
  • maxnorm: valor máximo de la norma

Ejemplo de uso

Ejemplo

import torch

# Crear tensor
x = torch.tensor([[2.0, 4.0, 6.0], [3.0, 6.0, 9.0]])

# Normalización de la norma L2 en dim=1, con norma máxima de 1
y = torch.renorm(x, p=2, dim=1, maxnorm=1)
print(y)

El resultado es:

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

Ejemplo

import torch

# Crear tensor
x = torch.tensor([[1.0, 2.0, 3.0]])

# Normalización de la norma L1 en dim=1
y = torch.renorm(x, p=1, dim=1, maxnorm=1)
print(y)

El resultado es:

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

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

Otras extensiones