pytorch

Deep learning library for tensors and dynamic neural networks

brewmacoslinux

Try with needOr install directly

About

Tensors and dynamic neural networks

python

Import PyTorch and create a tensor$ python -c "import torch; x = torch.tensor([1.0, 2.0, 3.0]); print(x)"

Build a simple neural network model$ python -c "import torch; import torch.nn as nn; model = nn.Linear(10, 5); print(model)"

Perform tensor operations and autograd

$ python -c "import torch; x = torch.tensor([1.0, 2.0], requires_grad=True); y = (x**2).sum(); y.backward(); print(x.grad)"