Closed juharris closed 4 years ago
Translate PyTorch's gather to Tensorflow.js' gather. I am trying to write a Plan in Python with PyTorch that uses tensor.select and I would like it translated so that I can use syft,js. Did I do this right?
tensor.select
syft,js
Added tests.
@Nolski I resolved the conflicts.
Description
Translate PyTorch's gather to Tensorflow.js' gather. I am trying to write a Plan in Python with PyTorch that uses
tensor.select
and I would like it translated so that I can usesyft,js
. Did I do this right?How has this been tested?
Added tests.
Checklist