更改theano張量向量中的值

Question

我有一個theano張量向量next_probs ，類似於[0.222, 0.34342, 0.41324, 0.1231, ...] ，它是以下函數的輸出：

next_probs = tensor.nnet.softmax(logit)

logit是一個向量，其尺寸與next_probs相同。

如何在next_probs向量next_probs一個特定值更改為1 ，將另一個特定值更改為0 ？

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是一個向量，該向量在該類中具有最高的概率，而其他則為0