pulsatrix::TopKResult Struct Reference

top_k()'s result: the selected values and their positions along the last dimension. More...

#include <top_k.hpp>

Collaboration diagram for pulsatrix::TopKResult:

Public Attributes
Tensor	values
	Shape (..., k): each row's k best values, best first.
Tensor	indices
	Shape (..., k): each value's index within its row, as a whole-number float.

Detailed Description

top_k()'s result: the selected values and their positions along the last dimension.

Member Data Documentation

indices

Tensor pulsatrix::TopKResult::indices

Shape (..., k): each value's index within its row, as a whole-number float.

values

Tensor pulsatrix::TopKResult::values

Shape (..., k): each row's k best values, best first.

---

The documentation for this struct was generated from the following file:

• include/pulsatrix/top_k.hpp