Skip to contents

Applies a linear transformation to the incoming data: \(y = xA^T + b\).

Usage

nnf_linear(input, weight, bias = NULL)

Arguments

input

\((N, *, in\_features)\) where * means any number of additional dimensions

weight

\((out\_features, in\_features)\) the weights tensor.

bias

optional tensor \((out\_features)\)