2022年3月24日
torch topk函数
摘要： 这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index。 用法 torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor) input：一个tens 阅读全文
posted @ 2022-03-24 10:37 Tomorrow1126 阅读(2459) 评论(0) 推荐(0) 编辑