Si quieren iterar a través de todos los axes podrían utilizar algo como esto y también les recomiendo utilizar los notebooks de VS Code para aprovechar la potencial del editor en auto completado y todo lo demás a no ser que necesiten una GPU o TPU que te brinda Google Colab.
¿Quieres ver más aportes, preguntas y respuestas de la comunidad? Crea una cuenta o inicia sesión.