Searched refs:select_nested (Results 1 – 2 of 2) sorted by relevance
160 fn select_nested() { in select_nested() function
427 Tensor select_nested(const Tensor& self, int64_t dim, int64_t index) { in select_nested() function