nnf_linear: Linear

View source: R/nnf-linear.R

nnf_linearR Documentation

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)


torch documentation built on May 29, 2024, 9:54 a.m.