2024年3月2日
Pytorch中张量的高级选择操作
摘要： 在某些情况下，我们需要用Pytorch做一些高级的索引/选择，所以在这篇文章中，我们将介绍这类任务的三种最常见的方法：torch.index_select, torch.gather and torch.take 我们首先从一个2D示例开始，并将选择结果可视化，然后延申到3D和更复杂场景。最后以表格 阅读全文
posted @ 2024-03-02 13:55 deephub 阅读(2) 评论(0) 推荐(0) 编辑