torchEasy Playground

PyTorch: dynamic neural networks with GPU acceleration by Meta

Getting started with torchRun locally
Install
pip install torch
Python CodeRun locally
Expected Output
# Expected output shown below
# (Run locally with: torch)

torch is a third-party package. PyTorch: dynamic neural networks with GPU acceleration by Meta. Install with: pip install torch

Challenge

Try modifying the code above to explore different behaviors. Can you extend the example to handle a new use case?