Module adv_indexing

Module adv_indexing 

Source
Expand description

Advanced indexing related tensor manuplications.

Currently, full support of advanced indexing is not available. However, it is still possible to index one axis by list.

Functionsยง

bool_select
Returns a new tensor, which indexes the input tensor along dimension axis using the boolean entries in mask.
bool_select_f
index_select
Returns a new tensor, which indexes the input tensor along dimension axis using the entries in indices.
index_select_f
take
Take elements from an array along an axis.
take_f