- all_moves[:, 0] = monster[:, t]
- all_moves[:, 1, 1:, :] = monster[:, t, :-1, :]
- all_moves[:, 2, :-1, :] = monster[:, t, 1:, :]
- all_moves[:, 3, :, 1:] = monster[:, t, :, :-1]
- all_moves[:, 4, :, :-1] = monster[:, t, :, 1:]
- a = F.one_hot(monster_actions[:, t], num_classes=5)[:, :, None, None]
+ all_moves[:, 0] = troll[:, t]
+ all_moves[:, 1, 1:, :] = troll[:, t, :-1, :]
+ all_moves[:, 2, :-1, :] = troll[:, t, 1:, :]
+ all_moves[:, 3, :, 1:] = troll[:, t, :, :-1]
+ all_moves[:, 4, :, :-1] = troll[:, t, :, 1:]
+ a = F.one_hot(troll_actions[:, t], num_classes=5)[:, :, None, None]