chenyu f3a50b4e40 fix broadcasted logic if there's 0 in shapes (#3097) ...

* fix broadcasted logic if there's 0 in shapes

should always expand into 0, not the other way around. fixed matmul with 0 in input shapes.
for forwards for now though, backward is more involved and would need to change 0 size shortcuts

* fix tests

2024-01-12 13:32:43 -05:00

..

__init__.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

test_indexing.py

fix broadcasted logic if there's 0 in shapes (#3097)

2024-01-12 13:32:43 -05:00