How to compute element-wise entropy of an input tensor in PyTorch

In this article, we are going to discuss how to compute the element-wise entropy of an input tensor in PyTorch, we can compute this by using torch.special.entr() method.

torch.special.entr() method

torch.special.entr() method computes the element-wise entropy, This method accepts a tensor as input and returns a tensor with the element-wise entropy of the input tensor. if the element is zero, or negative then entropy is also zero, or negative infinity respectively. before moving further let's see the syntax of the given method.

Syntax: torch.special.entr(tens)

Parameters:

• tens: This is our input tensor.

Returns: Returns the elements-wise entropy of an input tensor.

Example 1:

The following program is to understand how to compute the element-wise entropy of a 1D tensor.

# 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)

Output:

Example 2:

The following program is to know how to compute the element-wise entropy of a 2D tensor.

# 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)

Output: