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

函数功能

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

函数原型

add_matrix_multiply(self: ascendie.Network, arg0: ascendie.Tensor, arg1: ascendie.MatrixOperation, arg2: ascendie.Tensor, arg3: ascendie.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。

参数说明