Sort Elements of a Tensor in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 10:10:46

2K+ Views

To sort the elements of a tensor in PyTorch, we can use the torch.sort() method. This method returns two tensors. The first tensor is a tensor with sorted values of the elements and the second tensor is a tensor of indices of elements in the original tensor. We can compute the 2D tensors, row-wise and column-wise.StepsImport the required library. In all the following Python examples, the required Python library is torch. Make sure you have already installed it.Create a PyTorch tensor and print it.To sort the elements of the above-created tensor, compute torch.sort(input, dim). Assign this value to a new ... Read More

Compute Logarithm of Elements of a Tensor in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 10:08:01

1K+ Views

To compute the logarithm of elements of a tensor in PyTorch, we use the torch.log() method. It returns a new tensor with the natural logarithm values of the elements of the original input tensor. It takes a tensor as the input parameter and outputs a tensor.StepsImport the required library. In all the following Python examples, the required Python library is torch. Make sure you have already installed it.Create a tensor and print it.Compute torch.log(input). It takes input, a tensor, as the input parameter and returns a new tensor with the natural logarithm values of elements of the input.Print the tensor ... Read More

Get Data Type of a Tensor in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 10:06:20

14K+ Views

A PyTorch tensor is homogenous, i.e., all the elements of a tensor are of the same data type. We can access the data type of a tensor using the ".dtype" attribute of the tensor. It returns the data type of the tensor.StepsImport the required library. In all the following Python examples, the required Python library is torch. Make sure you have already installed it.Create a tensor and print it.Compute T.dtype. Here T is the tensor of which we want to get the data type.Print the data type of the tensor.Example 1The following Python program shows how to get the data ... Read More

Squeeze and Unsqueeze a Tensor in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 10:03:48

5K+ Views

To squeeze a tensor, we use the torch.squeeze() method. It returns a new tensor with all the dimensions of the input tensor but removes size 1. For example, if the shape of the input tensor is (M ☓ 1 ☓ N ☓ 1 ☓ P), then the squeezed tensor will have the shape (M ☓ M ☓ P).To unsqueeze a tensor, we use the torch.unsqueeze() method. It returns a new tensor dimension of size 1 inserted at specific position.StepsImport the required library. In all the following Python examples, the required Python library is torch. Make sure you have already installed ... Read More

Compute Histogram of a Tensor in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 10:02:22

2K+ Views

The histogram of a tensor is computed using torch.histc(). It returns a histogram represented as a tensor. It takes four parameters: input, bins, min and max. It sorts the elements into equal width bins between min and max. It ignores the elements smaller than the min and greater than the max.StepsImport the required library. In all the following Python examples, the required Python libraries are torch and Matplotlib. Make sure you have already installed them.Create a tensor and print it.Compute torch.histc(input, bins=100, min=0, max=100). It returns a tensor of histogram values. Set bins, min, and max to appropriate values according ... Read More

Find Mean Across Image Channels in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 10:00:45

2K+ Views

RGB images have three channels, Red, Green, and Blue. We need to compute the mean of the image pixel values across these image channels. For this purpose, we use the method torch.mean(). But the input parameter to this method is a PyTorch tensor. So, we first convert the image to the PyTorch tensor and then apply this method. It returns the mean value of all the elements in the tensor. To find the mean across the image channels, we set the parameter dim = [1, 2].StepsImport the required library. In all the following Python examples, the required Python libraries are ... Read More

Find K-th and Top K Elements of a Tensor in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 09:56:34

902 Views

PyTorch provides a method torch.kthvalue() to find the k-th element of a tensor. It returns the value of the k-th element of tensor sorted in ascending order, and the index of the element in the original tensor.torch.topk() method is used to find the top "k" elements. It returns the top "k" or largest "k" elements in the tensor.StepsImport the required library. In all the following Python examples, the required Python library is torch. Make sure you have already installed it.Create a PyTorch tensor and print it.Compute torch.kthvalue(input, k). It returns two tensors. Assign these two tensors to two new variables ... Read More

Compute Mean and Standard Deviation of a Tensor in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 09:52:59

6K+ Views

A PyTorch tensor is like a numpy array. The only difference is that a tensor utilizes the GPUs to accelerate numeric computations. The mean of a tensor is computed using the torch.mean() method. It returns the mean value of all the elements in the input tensor. We can also compute the mean row-wise and column-wise, providing suitable axis or dim.The standard deviation of a tensor is computed using torch.std(). It returns the standard deviation of all the elements in the tensor. Like mean, we can also compute the standard deviation, row or column-wise.StepsImport the required library. In all the following ... Read More

Element-wise Division on Tensors in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 09:51:34

5K+ Views

To perform element-wise division on two tensors in PyTorch, we can use the torch.div() method. It divides each element of the first input tensor by the corresponding element of the second tensor. We can also divide a tensor by a scalar. A tensor can be divided by a tensor with same or different dimension. The dimension of the final tensor will be same as the dimension of the higher-dimensional tensor. If we divide a 1D tensor by a 2D tensor, then the final tensor will a 2D tensor.StepsImport the required library. In all the following Python examples, the required Python ... Read More

Element-Wise Multiplication on Tensors in PyTorch

Shahid Akhtar Khan
Updated on 06-Nov-2021 09:49:57

14K+ Views

torch.mul() method is used to perform element-wise multiplication on tensors in PyTorch. It multiplies the corresponding elements of the tensors. We can multiply two or more tensors. We can also multiply scalar and tensors. Tensors with same or different dimensions can also be multiplied. The dimension of the final tensor will be same as the dimension of higher-dimensional tensor. Element-wise multiplication on tensors are also known as Hadamard product.StepsImport the required library. In all the following Python examples, the required Python library is torch. Make sure you have already installed it.Define two or more PyTorch tensors and print them. If ... Read More

Advertisements