Add a `dim`/`axis` arg to support batched tensors - Githubissues

bamos / block

An intelligent block matrix library for numpy, PyTorch, and beyond.

Apache License 2.0

297 stars 33 forks source link

Add a `dim`/`axis` arg to support batched tensors #17

Open bamos opened 5 years ago