Skip to content

Instantly share code, notes, and snippets.

@sailfish009
Last active October 14, 2019 14:54
Show Gist options
  • Save sailfish009/08cbd53adf9f6cbfb0f8bc804894b8e3 to your computer and use it in GitHub Desktop.
Save sailfish009/08cbd53adf9f6cbfb0f8bc804894b8e3 to your computer and use it in GitHub Desktop.
https://pytorch.org/docs/stable/index.html
http://blog.ezyang.com/2019/05/pytorch-internals/
https://www.youtube.com/playlist?list=PLZbbT5o_s2xrfNyHZsM6ufI0iZENK9xgG
torch.nn.Linear(W_target.size(0), 1)
Applies a linear transformation to the incoming data: y = xA^T + b
torch.unsqueeze(input, dim, out=None) → Tensor :
zero_grad() : Sets gradients of all model parameters to zero.
torch.randn() → Tensor :
Returns a tensor filled with random numbers from a normal distribution with mean 0 and variance 1 (also called the standard normal distribution).
torch.tensor() :
A torch.Tensor is a multi-dimensional matrix containing elements of a single data type.
torch.mm(input, mat2, out=None) → Tensor
Performs a matrix multiplication of the matrices input and mat2.
If input is a (n×m) tensor, mat2 is a (m×p) tensor, out will be a (n×p) tensor.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment