Support slicing with fewer axes than tensor input

[rdragos]

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