targets = maze.stationary_densities(
mazes[:, : task.height * task.width].view(-1, task.height, task.width),
policies.view(-1, 4, task.height, task.width),
targets = maze.stationary_densities(
mazes[:, : task.height * task.width].view(-1, task.height, task.width),
policies.view(-1, 4, task.height, task.width),