功能描述
算子功能
矩阵内积算子。
算子公式
2D情形下,矩阵内积计算公式如下:

高维Tensor将被展平为2D Tensor进行计算。
参数 |
参数说明 |
|---|---|
n |
batch数。 |
ic |
input channel,输入通道数。 |
oc |
output channel,输出通道数。 |
src |
源输入Tensor。 |
weights |
权重Tensor。 |
bias |
偏置Tensor。 |
dst |
输出结果Tensor。 |
父主题: Inner Product