Pytorch

[[nn.ModuleList]] 存储不同 module，并将每个 module 的 parameters 添加到网络之中的容器

[[torch.nn]]

torch.topk

[[torch.gather]]

[[Pytorch/tensor]]

torch.argsort(*input*, *dim=-1*, *descending=False*, *stable=False*) → [Tensor]

• 按 input dim 维度排序

• 返回 input 对应元素的排序结果

https://www.zhihu.com/question/352525266/answer/3395318281 pytorch 架构