Baremetal-NN
Baremetal-NN API documentation
|
Applies a linear transformation to the incoming data
y = x @ w.T + b
y | the output tensor of shape (1, out_features) |
x | tnput tensor of shape (1, in_features) |
w | weight tensor of shape (out_features, in_features) |
b | bias tensor of shape (1, out_features), or NULL if no bias is applied |