search

BaguaSys / bagua-core

Core communication lib for Bagua.
https://bagua-tutorials.kwai-seattle.com/
MIT License
47 stars 8 forks source link

Pass torch tensor directly to make interface easier #6

Closed NOBLES5E closed 3 years ago

NOBLES5E commented 3 years ago

https://github.com/BaguaSys/bagua-core/blob/b9528617343fd671d57b29890606025735e740f8/bagua-core-py/src/lib.rs#L85-L91

create-issue-branch[bot] commented 3 years ago

Branch issue-6-Pass_torch_tensor_directly_to_make_interface_easier created!