nnf_linear {torch}

Linear

Description

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)