[Refactor] Added axes() function in impl.py (#2594)
[Refactor] Added axes() function in impl.py (#2594)

* Added axes() function in impl.py

* Auto Format

Co-authored-by: Taichi Gardener <52844ac7498fba84f633481a7c3ae54491811c81@gmail.com>
yuanming-hu/taichi,yuanming-hu/taichi,yuanming-hu/taichi,yuanming-hu/taichi