示例:输入：A:{c=12 h=64 w=26}，B:{c=12 h=26 w=64}输出：C:{c=1 h=64 w=64}输出结果被简化为了 A[0] * B[0]，希望支持channel维度，期望结果应该为 C:{c=12 h=64 w=64}
00jrzges1#
手动把Gemm修改为MatMul即可。做法：用文本编辑器打开.param文件，找到这个Gemm所在的行，把这一行最开始的那个“Gemm”手动修改为“MatMul”。
1条答案
按热度按时间00jrzges1#
手动把Gemm修改为MatMul即可。
做法：用文本编辑器打开.param文件，找到这个Gemm所在的行，把这一行最开始的那个“Gemm”手动修改为“MatMul”。