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

Enable ATen implementation of some NN functions and Variable methods

Browse Source
This commit is contained in:
Sam Gross
2017-10-16 11:34:58 -07:00
committed by Edward Z. Yang
parent a385979677
commit 5989b05ecc
6 changed files with 182 additions and 624 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/csrc/utils/python_arg_parser.cpp
View File

@ -78,6 +78,9 @@ bool FunctionParameter::check(PyObject* obj) {
if (PyTuple_Check(obj) || PyList_Check(obj)) {
return true;
}
if (optional && obj == Py_None) {
return true;
}
// if a size is specified (e.g. IntList[2]) we also allow passing a single int
return size > 0 && THPUtils_checkLong(obj);
}