liginity liginity
liginity
/ gradient-descent-on-linear-mapping.py
Last active
July 3, 2023 01:58
use pytorch functionality to realize gradient descent on linear mapping (or linear regression).
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """use pytorch in this example. | |
| input: dimension = (16,) | |
| output: dimension = (4,) | |
| matrix of linear mapping: dimension = (16, 4) | |
| """ | |
| import torch | |
| from torch import nn |