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

Implement Variable.new (#4080)

Browse Source
This commit is contained in:
Sam Gross
2017-12-11 15:45:43 -05:00
committed by GitHub
parent 8612b0bbd8
commit aeb7a3668d
21 changed files with 313 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tools/autograd/gen_python_functions.py
View File

@ -113,7 +113,10 @@ def create_python_bindings(
typename = 'Tensor'
unpack = unpack_methods.get(typename, typename.lower())
actuals.append('r.{}({})'.format(unpack, arg_idx))
expr = 'r.{}({})'.format(unpack, arg_idx)
if typename == 'Storage &':
expr = '*' + expr
actuals.append(expr)
dispatch_type = typename
if dispatch_type == 'Tensor':
dispatch_type = 'const Tensor &'