Cómo calcular la entropía de elementos de un tensor de entrada en PyTorch

En este artículo, vamos a discutir cómo calcular la entropía de elementos de un tensor de entrada en PyTorch, podemos calcular esto usando el método torch.special.entr().

método torch.special.entr()

El método torch.special.entr() calcula la entropía de los elementos. Este método acepta un tensor como entrada y devuelve un tensor con la entropía de los elementos del tensor de entrada. si el elemento es cero o negativo, entonces la entropía también es cero o infinito negativo, respectivamente. antes de continuar, veamos la sintaxis del método dado.

Sintaxis: torch.special.entr(decenas)

Parámetros:

• decenas: Este es nuestro tensor de entrada.

Devoluciones: Devuelve la entropía de los elementos de un tensor de entrada.

Ejemplo 1:

El siguiente programa es para comprender cómo calcular la entropía de elementos de un tensor 1D.

# import torch libraries
import torch
  
# creating a 1D tensor
tens = torch.tensor([4, 5, 0, -5, -4])
  
# Display tensor
print("\n\nInput Tensor: ", tens)
  
# compute the element-wise entropy of 
# input tensor
entr = torch.special.entr(tens)
  
# Display result
print("\n\nComputed Entropy: ", entr)

Producción:

 

Ejemplo 2:

El siguiente programa es para saber cómo calcular la entropía de elementos de un tensor 2D.

# import torch libraries
import torch
  
# creating a 2D tensor
tens = torch.tensor([[1, 2, -3],
                     [0, -3, 2],
                     [-2, 0, -3]])
  
# Display tensor
print("\n Input Tensor: \n", tens)
  
# compute the element-wise entropy of
# input tensor
entr = torch.special.entr(tens)
  
# Display result
print("\n Computed Entropy: \n", entr)

Producción: