python (65.2k questions)

javascript (44.3k questions)

reactjs (22.7k questions)

java (20.8k questions)

c# (17.4k questions)

html (16.3k questions)

r (13.7k questions)

android (13k questions)

Questions - tensor

How can I find multiple maximum indices of a torch tensor?

If I have a tensor which has multiple maximum values how can I get all the indices of maximum value. I have tried torch.argmax(tensor) but it only gives me the first index. >>> a_list = [3,23...
test-img

H.Jamil

python

pytorch

tensor

Votes: 0

Answers: 1

Latest Answer

Find the maximum value, then find all elements with that value. (x == torch.max(x)).nonzero() Note: nonzero may also be called with as_tuple=True, which may be helpful.
test-img

Mateen Ulhaq

Torch-wise masking instead of element-wise

I would like to create a mask for a tensor based on the existence of another tensor within the first. This is easy to do if I am checking for a scalar, but I could not find a way to do it tensor-wise ...
test-img

Sydney

python

pytorch

tensor

Votes: 0

Answers: 0

switch indices in 4th order tensor with python

I would like to replicate the following operation: import numpy as np a = np.eye(3) A = np.einsum("ij,kl->ikjl", a,a) I have the following: def IndexPermutation(stringWithTensorIndices, ...
test-img

user17744436

python

numpy

tensor

Votes: 0

Answers: 1

Latest Answer

I understand that what you want to do is provide some input stringWithTensorIndices with values like "ijkl" or "jlki" to determine how the summation will be achieved. Then the logi...
test-img

Paddy Alton

How to Reverse Order of Rows in a Tensor

I'm trying to reverse the order of the rows in a tensor that I create. I have tried with tensorflow and pytorch. Only thing I have found is the torch.flip() method. This does not work as it reverses n...
test-img

LearnerHere

python

tensorflow

pytorch

tensor

Votes: 0

Answers: 1

Latest Answer

According to documentation torch.flip has argument dims, which control what axis to be flipped. In this case torch.flip(tensor_a, dims=(0,)) will return expected result. Also torch.flip(tensor_a) will...
test-img

draw

Posts

Questions

Blogs

Jobs