- probas[:, 1:, :] = pred_probas[:, :-1, :] * policies[:, 0, :-1, :]
- probas[:, :-1, :] = pred_probas[:, 1:, :] * policies[:, 1, 1:, :]
- probas[:, :, 1:] = pred_probas[:, :, :-1] * policies[:, 2, :, :-1]
- probas[:, :, :-1] = pred_probas[:, :, 1:] * policies[:, 3, :, 1:]
+ probas[:, 1:, :] += pred_probas[:, :-1, :] * policies[:, 3, :-1, :]
+ probas[:, :-1, :] += pred_probas[:, 1:, :] * policies[:, 2, 1:, :]
+ probas[:, :, 1:] += pred_probas[:, :, :-1] * policies[:, 1, :, :-1]
+ probas[:, :, :-1] += pred_probas[:, :, 1:] * policies[:, 0, :, 1:]