mirror of
https://github.com/pytorch/pytorch.git
synced 2026-10-28 18:56:51 +08:00
The fix-cpp-build branch is a major restructuring that: - Drops Exhale (which auto-generated API docs but caused OOM in CI) in favor of hand-written RST pages using Breathe directives - Reorganizes notes into a hierarchical API reference with proper categories - Adds new content: XPU docs, stable ABI docs, DeviceGuard/Stream docs - Always runs Doxygen instead of gating on an env var Pull Request resolved: https://github.com/pytorch/pytorch/pull/174096 Approved by: https://github.com/janeyx99, https://github.com/seemethere, https://github.com/albanD
1.6 KiB
1.6 KiB
myst
| myst | ||||||
|---|---|---|---|---|---|---|
|
Linear Layers
Linear layers apply affine transformations to input data: y = xW^T + b.
They are the building blocks of fully-connected networks and are used for
feature transformation, classification heads, and projection layers.
- Linear: Standard fully-connected layer
- Bilinear: Bilinear transformation of two inputs
- Identity: Pass-through layer (useful for skip connections)
- Flatten/Unflatten: Reshape tensors between convolutional and linear layers
Linear
:members:
:undoc-members:
:members:
:undoc-members:
Example:
auto linear = torch::nn::Linear(torch::nn::LinearOptions(784, 256).bias(true));
auto output = linear->forward(input); // input: [N, 784]
Bilinear
:members:
:undoc-members:
:members:
:undoc-members:
Identity
:members:
:undoc-members:
:members:
:undoc-members:
Flatten
:members:
:undoc-members:
:members:
:undoc-members:
Unflatten
:members:
:undoc-members:
:members:
:undoc-members: