向量化将函数应用于三维数组我试图将一个函数应用于3d torch张量,同时将该函数应用于通过3d torch张量的轴1读取的2d张量 例如,我有一个(51, 128, 20100)形状的torch张量(一个名为autoenco ...2024-10-03 已阅读: n次