tf-encrypted / moose

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

Support slicing with fewer axes than tensor input #1065

Open rdragos opened 2 years ago

rdragos commented 2 years ago

Currently we need to specify for all axes how we do slicing. In numpy, and other higher level languages one can simply do x[:1].

We'd like add the numpy slicing convention in the runtime, see for more details: https://numpy.org/doc/stable/user/basics.indexing.html