2020年3月14日
PyTorch常用函数：torch.topk; torch.index_select; torch.lt
摘要： torch.topk torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor) pytorch中文官网文档：http://www.mamicode.com/info-deta 阅读全文
posted @ 2020-03-14 21:21 lqchen 阅读(6422) 评论(0) 推荐(0)