|
Baremetal-NN
Baremetal-NN API documentation
|
| void nn_linear_f32 | ( | Tensor2D_F32 * | y, |
| const Tensor2D_F32 * | x, | ||
| const Tensor2D_F32 * | weight, | ||
| const Tensor1D_F32 * | bias | ||
| ) |
Linear neural network layer.
nn_linear_f32
y[i][j] = x[i][k] * weight[j][k] + bias[j]
| y | The result tensor. |
| x | The input tensor. |
| weight | The weight tensor. |
| bias | The bias tensor. |