使用idex数组收集切片我有一个大小为(BxHxWx200)的输入张量params,还有一个大小为(Bx1000x2)的索引数组idxidx包含1000个不同位置的param张量,即最后一个轴(大小为2的轴)是HxW位置,对 ...2024-09-29 已阅读: n次