摘要: 前言 在某些情况下,我们需要用Pytorch做一些高级的索引/选择,所以在这篇文章中,将介绍这类任务的三种最常见的方法:torch.index_select, torch.gather and torch.take 本文转载自Deephub Imba 作者:Oliver S 仅用于学术分享,若侵权请 阅读全文
posted @ 2024-03-05 01:05 CV技术指南(公众号) 阅读(87) 评论(0) 推荐(0) 编辑