[MindSpore]训练LeNet5报错x1_col,x2_row,For ‘MatMul‘, the input dimensions must be equal
mindspore1.6.1 cpu版本
报错信息:
ValueError: For 'MatMul', the input dimensions must be equal, but got 'x1_col': 256 and 'x2_row': 400. And 'x' shape [8, 256](transpose_a=False), 'y' shape [120, 400](transpose_b=True).
这句报错描述的是我们在MatMul算子中进行张量的计算时
出现了不符合线性代数中乘法运
原创
2022-04-29 14:04:03 ·
1943 阅读 ·
0 评论