[arch-commits] Commit in python-pytorch/trunk (test.py)

[Sven-Hendrik Haase via arch-commits]

    Date: Saturday, December 12, 2020 @ 11:41:46
  Author: svenstaro
Revision: 773268

Add simple test case

Added:
  python-pytorch/trunk/test.py

---------+
 test.py |    5 +++++
 1 file changed, 5 insertions(+)

Added: test.py
===================================================================
--- test.py                             (rev 0)
+++ test.py     2020-12-12 11:41:46 UTC (rev 773268)
@@ -0,0 +1,5 @@
+import torch
+
+d = torch.device('cuda')
+a = torch.rand(1, 2).to(d)
+print(a + 0)