Tensor multiplication

vinhkhuc · Feb 13, 2017 · 2e26abe · 2e26abe
1 parent 6cc505c
commit 2e26abe
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/0_multiply.py b/0_multiply.py
@@ -0,0 +1,6 @@
+import torch
+
+a = torch.IntTensor([2, 3, 4])
+b = torch.IntTensor([3, 4, 5])
+m = a * b  # element-wise product
+print(m.numpy())  # convert to the numpy array [ 6 12 20]