search

tf-encrypted / moose

Secure distributed dataflow framework for encrypted machine learning and data processing
Apache License 2.0
57 stars 15 forks source link

Add axes argument to Transpose Rust Kernel. #413

Open yanndupis opened 3 years ago

yanndupis commented 3 years ago

In the Python kernel, we support axes argument (see numpy for examples) but not in the Rust kernel.

yanndupis commented 2 years ago

It's still hasn't been implemented. It's not required by any model right now. But it's a feature supported by numpy, pytorch etc.

yanndupis commented 2 years ago

Also it's an optional arg on in the edsl but not used by the rust kernel. So we should complete the rust kernel or remove the arg in the eds.