shlogg · Early preview
Super Kai (Kazuya Ito) @superkai_kazuya

Understanding PyTorch's Linear Layer And Its Arguments

Exploring PyTorch's Linear Layer: Initialization & Usage

Buy Me a Coffee☕
*Memos:

My post explains Linear layer(Fully-connected Layer).
My post explains manual_seed().
My post explains requires_grad.

Linear() can get the 1D or more D tensor of the zero or more elements computed by Affine transformation from the 1D or more D tensor of zero or more elements as shown below:
*Memos:

The 1st argument for initialization is in_features(Required-Type:float or complex). *It must be 0 <= x.
The 2nd argument for initialization is out_features(Required-Default:False-Type:float):
*Memos:

It must be 0 <= x.
0 is possible but warning occurs.


The 3rd argumen...