def add_matrix_multiply(self: Network, input0: Tensor, type0: MatrixOperation, input1: Tensor, type1: MatrixOperation) -> MatrixMultiplyLayer

函数功能

将矩阵乘法层添加到网络中。

函数原型

add_matrix_multiply(self: ascendie.pyascendie.Network, arg0: ascendie.pyascendie.Tensor, arg1: ascendie.pyascendie.MatrixOperation, arg2: ascendie.pyascendie.Tensor, arg3: ascendie.pyascendie.MatrixOperation) -> AscendIE::MatrixMultiplyLayer

约束说明

• input0, input1需为有效输入，维度范围[1,8]，每一维不可为0，也不可小于-1。
• 当input0,input1为矩阵或向量时，将会计算input0与input1的内积，方法中将会校验给定输入是否满足内积运算的基本法则。（如转置后input0的最后一维需要与input1的倒数第二维取值相同）
• 暂不支持一个输入是一维向量，另一个输入是多维向量(维度大于2)的情况，例如，[4] * [4, 5, 6] 或[6, 5, 4] * [4]
• 当前不支持动态shape。

参数说明