bamos / block

An intelligent block matrix library for numpy, PyTorch, and beyond.
Apache License 2.0
297 stars 33 forks source link

new pytorch datatype support #14

Closed hbgtjxzbbx closed 6 years ago

hbgtjxzbbx commented 6 years ago

The new pytorch(0.4) change its data type, so some fixes need to be done.