chore: add PyTorch

seanwu1105 · Nov 3, 2022 · 6445155 · 6445155
1 parent 6fdf9fb
commit 6445155
Show file tree

Hide file tree

Showing 4 changed files with 372 additions and 3 deletions.
diff --git a/.pylintrc b/.pylintrc
@@ -220,7 +220,7 @@ contextmanager-decorators=contextlib.contextmanager
 # List of members which are set dynamically and missed by pylint inference
 # system, and so shouldn't trigger E1101 when accessed. Python regular
 # expressions are accepted.
-generated-members=
+generated-members=torch.*
 
 # Tells whether to warn about missing members when the owner of the attribute
 # is inferred to be None.

diff --git a/main.py b/main.py
@@ -0,0 +1,6 @@
+import torch
+
+x = torch.rand(5, 3)
+print(x)
+
+print(torch.cuda.is_available())