topk_last_dim
ferritin_
plms
0.3.3-dev
In ferritin_
plms::
featurize::
utilities
ferritin_plms
::
featurize
::
utilities
Function
topk_
last_
dim
Copy item path
Source
pub fn topk_last_dim(xs: &Tensor, topk:
usize
) -> Result<(Tensor, Tensor)>