This document is relevant for: Trn2, Trn3
nki.language.ds#
- nki.language.ds(start, size)[source]#
Create a dynamic slice for tensor indexing.
- Parameters:
start – the start index of the slice.
size – the size of the slice.
- Returns:
a dynamic slice object for use in tensor indexing.
This document is relevant for: Trn2, Trn3