mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
548c460bf1
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/129758 Approved by: https://github.com/ezyang
18 lines
414 B
Python
18 lines
414 B
Python
# Owner(s): ["module: unknown"]
|
|
|
|
import argparse
|
|
|
|
import torch
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
|
|
# Required args. Raises error if they aren't passed.
|
|
parser.add_argument("--foo", help="foo", required=True)
|
|
parser.add_argument("--bar", help="bar", required=True)
|
|
_ = parser.parse_args()
|
|
|
|
x = torch.ones((3, 3), requires_grad=True)
|
|
(3 * x).sum().backward()
|