2025年6月18日
gather算子大不同
摘要： gather算子大不同 本文通过2个实际的案例，演示了一下gather算子在MindSpore框架下PyTorch框架下的异同点。两者的输入都是tensor-axis-index，一个是输入顺序上略有区别，另一个是对于输入的张量索引维度的要求。在PyTorch中，如果我们要实现类似于MindSpore中的gather功能，需要手动对输入索引的维度操作一下。 阅读全文
posted @ 2025-06-18 15:06 DECHIN 阅读(123) 评论(0) 推荐(0)