miniworld.utils.output_visualize.visualize_2D_tensor

miniworld.utils.output_visualize.visualize_2D_tensor(tensor, saving_path='test.png')