[Diagram] How to use torch.gather() Function in PyTorch with Examples
Introduction
In this article, we will see how to use torch.gather() function in PyTorch. We shall first understand for what purpose the gather() function is used in PyTorch, along with its syntax. Finally, we will cover a few examples so that the concept can be understood easily by beginners.
What is torch.gather() function in PyTorch
As the name suggests, torch.gather() function is used to create a new tensor by gathering elements from an input tensor along a specific dimension and from specific indices.