diff options
Diffstat (limited to 'torch/nn/modules/utils.py')
-rw-r--r-- | torch/nn/modules/utils.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/torch/nn/modules/utils.py b/torch/nn/modules/utils.py index feb6c56d98..c3dc5126e6 100644 --- a/torch/nn/modules/utils.py +++ b/torch/nn/modules/utils.py @@ -1,6 +1,7 @@ import collections from itertools import repeat + def _ntuple(n): def parse(x): if isinstance(x, collections.Iterable): |