MatrixMultiplyLayer *AddMatrixMultiply(Tensor *input0, MatrixOperation type0, Tensor *input1, MatrixOperation type1) noexcept;

函数功能

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

函数原型

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。