projects
/
python.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[python.git]
/
poly.py
diff --git
a/poly.py
b/poly.py
index
ae9f88e
..
818742b
100755
(executable)
--- a/
poly.py
+++ b/
poly.py
@@
-5,6
+5,8
@@
# Written by Francois Fleuret <francois@fleuret.org>
# Written by Francois Fleuret <francois@fleuret.org>
+######################################################################
+
import torch
def pol_prod(a, b):
import torch
def pol_prod(a, b):
@@
-22,8
+24,10
@@
def pol_prim(a):
n[1:] = a / n[1:]
return n
n[1:] = a / n[1:]
return n
-a = torch.tensor([1., 2., 3.])
-b = torch.tensor([2., 5.])
+######################################################################
-print(pol_prod(a, b))
-print(pol_prim(b))
+if __name__ == '__main__':
+ a = torch.tensor([1., 2., 3.])
+ b = torch.tensor([2., 5.])
+ print(pol_prod(a, b))
+ print(pol_prim(b))