将矩阵乘法层添加到网络中。
MatrixMultiplyLayer *AddMatrixMultiply(Tensor *input0, MatrixOperation type0, Tensor *input1, MatrixOperation type1) noexcept;
参数名 |
输入/输出 |
说明 |
---|---|---|
input0 |
输入 |
第一个输入张量(通常为A)。 |
type0 |
输入 |
应用于 input0 的操作。矩阵操作类型,详见enum class MatrixOperation。 |
input1 |
输入 |
第二个输入张量(通常为B)。 |
type1 |
输入 |
应用于 input1 的操作。矩阵操作类型,详见enum class MatrixOperation。 |
返回新的矩阵乘法层,如果添加失败时返回nullptr。