Closed rdspring1 closed 2 weeks ago
The index argument does not have to be a TensorView for the select operation.
index
TensorView
select
Reference: NVF_API TensorView* select(TensorView* tv, int64_t dim, Val* index);
NVF_API TensorView* select(TensorView* tv, int64_t dim, Val* index);
!test
rdspring1
commented
2 weeks ago rdspring1
commented
2 weeks ago
The
indexargument does not have to be aTensorViewfor theselectoperation.Reference:
NVF_API TensorView* select(TensorView* tv, int64_t dim, Val* index);