poli.objective_repository.super_mario_bros.level_utils.levels_to_onehot#
- poli.objective_repository.super_mario_bros.level_utils.levels_to_onehot(levels: ndarray, n_sprites: int = 11) ndarray #
Transforms an array [b, w, h] of integers into a one-hot array [b, n_sprites, w, h].