2024年7月15日
torch topk 使用
摘要： 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 阅读(642) 评论(0) 推荐(0)
C++ tuple元组的基本用法
摘要： C++ tuple元组的基本用法 https://blog.csdn.net/sevenjoin/article/details/88420885 阅读全文
posted @ 2024-07-15 11:20 michaelchengjl 阅读(14) 评论(0) 推荐(0)