Imagine I’m doing RL, and I have per-state values represented as a tensor:
V = torch.IntTensor(2, 3) V.zero_()
Then I have a state represented as another tensor:
s = torch.IntTensor([1, 1])
… and I want to increment the corresponding value in V by 1, something like:
V[s] += 1 , or V.select(s) += 1, or
thoughts on how to do this? (I’d prefer not to have to write
V[s, s, s, ...] += 1 ideally). Happy to convert the index tensor to a LongTensor, if that will help?