all_max = tf.convert_to_tensor([[4, 2, 3], [3, 4, 5]], dtype=tf.float32)
Как получить индекс элемента [3,4,5] из тензорного массива all_max?
В списке мы просто используем list.index(element), чтобы получить индекс для элемента, присутствующего в списке.
Спасибо
я нашел это на: 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]))
@vishakraj проверьте это coderhelper.com/a/42190780/13319197
@vishakraj, надеюсь, это поможет тебе