search

google-research / dataclass_array

Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)
https://dataclass-array.readthedocs.io
Apache License 2.0
48 stars 3 forks source link

Fix broadcasting in pytorch #76

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Fix broadcasting in pytorch