Получить индекс от тензора tensorflow

avatar
vishak raj
9 августа 2021 в 07:00
46
1
0
all_max = tf.convert_to_tensor([[4, 2, 3], [3, 4, 5]], dtype=tf.float32)

Как получить индекс элемента [3,4,5] из тензорного массива all_max?

В списке мы просто используем list.index(element), чтобы получить индекс для элемента, присутствующего в списке.

Спасибо

Источник

Ответы (1)

avatar
H Sa
9 августа 2021 в 07:30
0

я нашел это на: https://www.py4u.net/discuss/147615 у меня сработало

Чтобы найти индекс элемента тензора 2d/3d, преобразуйте его в 1d #ie example.view(количество элементов)

Пример:

mat=torch.tensor([[1,2],[4,3])
#to find index of 2
five = 2
mat=mat.view(4)
numb_of_col = 4
for o in range(numb_of_col):
   if mat[o] == five:
   print(torch.tensor([o]))    
H Sa
9 августа 2021 в 07:54
0

@vishakraj проверьте это coderhelper.com/a/42190780/13319197

H Sa
9 августа 2021 в 07:55
0

@vishakraj, надеюсь, это поможет тебе