nn.Linear

Software / App

A PyTorch module that implements a linear transformation (fully connected layer) of the input data, typically in the form of y = Wx + b.

Mentioned in 1 video