How Pytorch Tensor get the index of specific value

python pytorch
Han Bing picture Han Bing · Dec 18, 2017 · Viewed 49.5k times · Source

In python list, we can use list.index(somevalue). How can pytorch do this?
For example: 

    a=[1,2,3]
    print(a.index(2))

Then, 1 will be output. How can a pytorch tensor do this without converting it to a python list?

Answer

Manuel Lagunas picture Manuel Lagunas · Dec 18, 2017

I think there is no direct translation from list.index() to a pytorch function. However, you can achieve similar results using tensor==number and then the nonzero() function. For example:

t = torch.Tensor([1, 2, 3])
print ((t == 2).nonzero())

This piece of code returns

1

[torch.LongTensor of size 1x1]

Related questions

How to check if pytorch is using the GPU?

I would like to know if pytorch is using my GPU. It's possible to detect with nvidia-smi if there is any activity from the GPU during the process, but I want something written in a python script. Is there a …

Read More
How to initialize weights in PyTorch?

How to initialize the weights and biases (for example, with He or Xavier initialization) in a network in PyTorch?

Read More
How does the "view" method work in PyTorch?

I am confused about the method view() in the following code snippet. class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2,2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = …

Read More