2025年7月17日
PyTorch中的take_along_dim
摘要： PyTorch中的take_along_dim 接前面一篇take_along_axis的文章，本文主要介绍在PyTorch框架下，功能基本一样的函数take_along_dim。二者除了命名和一些关键词参数不一致之外，用法是一样的。需要注意的是，两者都要求输入的数组和索引数组维度数量一致。在特定场景下，需要手动进行扩维。 阅读全文
posted @ 2025-07-17 10:18 DECHIN 阅读(280) 评论(0) 推荐(0)