Изменение значений в тензорной Theano вектора

Question

У меня есть тензор Theano Вектор next_probs что-то вроде [0.222, 0.34342, 0.41324, 0.1231, ...], который является выходом следующей функции:

next_probs = tensor.nnet.softmax(logit) 

logit представляет собой вектор, который имеет ту же размерность, next_probs.

Как я могу изменить одно конкретное значение на 1, а остальные на 0 в next_probs?

Answer 1

Какое значение конкретно вы хотите изменить? если вы просто хотите иметь вектор, которые имеют ту же размерность, как next_probs вы можете использовать zeros и set_subtensor как ниже

ret = T.zeros(next_probs.shape) 
ret = T.set_subtensor(ret[index],1) 

Если вы хотите использовать эту функцию в модели классификации, и вам нужен класс с наибольшей вероятностью в next_probs стать 1 и другие стали 0 это мой другой ответ:

ret = T.zeros(next_probs.shape) 
ret = T.set_subtensor(ret[T.argmax(next_probs)],1) 

ret вектор с 1 в классе с наибольшей вероятностью, и 0 для остальных