Kevin Herro bd6a0c90a0 add Tensor.split (#2750) ...

* add Tensor.split (#2677)

* fix mypy errors

* add list support for Tensor.split

* fix ruff comments

* match tensor.split api

* simplify split and test_split

---------

Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com>

2024-01-01 22:09:04 -08:00

81 KiB

Raw Blame History

View Raw