摘要: torch topk 使用 这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index。 用法 torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor 阅读全文
posted @ 2024-07-15 11:33 michaelchengjl 阅读(4) 评论(0) 推荐(0) 编辑
摘要: C++ tuple元组的基本用法 https://blog.csdn.net/sevenjoin/article/details/88420885 阅读全文
posted @ 2024-07-15 11:20 michaelchengjl 阅读(3) 评论(0) 推荐(0) 编辑