PyTorch Max: Get Maximum Value Of A PyTorch Tensor

PyTorch Max - Use PyTorch's max operation to calculate the max of a PyTorch tensor

Type: FREE   By: Sebastian Gutierrez, AIWorkbox.com Instructor Sebastian Gutierrez   Duration: 1:56   Technologies: PyTorch, Python

Page Sections: Video  |  Code  |  Transcript


< > Code:

You must be a Member to view code

Access all courses and lessons, gain confidence and expertise, and learn how things work and how to use them.

    or   Log In


Transcript:

This video will show you how to use PyTorch’s max operation to calculate the max of a PyTorch tensor.


First, we import PyTorch.

import torch


Then we check what version of PyTorch we are using.

print(torch.__version__)

We are using PyTorch version 0.4.1.


Let’s next create the tensor we’ll use for the PyTorch max operation example.

tensor_max_example = torch.tensor(
[
  [
      [1, 1, 1],
      [2, 2, 2],
      [3, 3, 3]
  ],
  [
      [4, 4, 4],
      [5,50, 5],
      [6, 6, 6]
  ]
]
)

So here we have torch.tensor which creates a floating tensor.

We have a 2x3x3 tensor that we’re going to assign to the Python variable tensor_max_example.


Visually, we can see that the max is going to be the number 50.


Let’s print the tensor_max_example Python variable to see what we have.

print(tensor_max_example)

We see that it’s a PyTorch tensor, we see that it is 2x3x3, and we still visually see that the max is going to be 50.


Next, let’s calculate the max of a PyTorch tensor using PyTorch tensor’s max operation.

tensor_max_value = torch.max(tensor_max_example)

So torch.max, we pass in our tensor_max_example, and we assign the value that’s returned to the Python variable tensor_max_value.


Let’s print the tensor_max_value variable to see what we have.

print(tensor_max_value)

We see that the max value is 50.


The one thing to note is that PyTorch returns the max value inside of a 0-dimensional tensor with the value of 50 inside of it.


Let’s double check to see what it is by using Python’s type operation.

type(tensor_max_value)

We can see that it’s the class of torch.Tensor.

So it is a PyTorch tensor that is returned with the max value inside of it.


Let’s use PyTorch’s item operation to get the value out of the 0-dimensional tensor.

torch.max(tensor_max_example).item()

And now, we get the number 50.


And just to make sure it’s a number, let’s use the Python type operation to check that this is actually a number.

type(torch.max(tensor_max_example).item())

And we see that it is a Python of class "int" for integer.


Perfect! We were able to use PyTorch’s max operation to calculate the max of a PyTorch tensor.



Back to deep learning tutorial lesson list