frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add torch.* functions

Browse Source
This commit is contained in:
Adam Paszke
2016-05-09 19:14:40 +02:00
parent d2738ca25e
commit 449ac4ca2a
11 changed files with 917 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@ -26,7 +26,7 @@ for template in templates:
generated += '\n{}\n'.format(new_content)
with open(in_init, 'r') as i:
header = i.read()
header, _, suffix = i.read().partition('# GENERATED CODE GOES HERE')
with open(out_init, 'w') as f:
f.write("""
@ -39,6 +39,7 @@ with open(out_init, 'w') as f:
""")
f.write(header)
f.write(generated)
f.write(suffix)
cwrap_src = ['torch/csrc/generic/TensorMethods.cwrap.cpp']
for src in cwrap_src: