update collections.abc import
parent
db6128db0f
commit
94ca140b67
|
@ -3,13 +3,13 @@
|
|||
Hacked together by / Copyright 2020 Ross Wightman
|
||||
"""
|
||||
from itertools import repeat
|
||||
from torch._six import container_abcs
|
||||
import collections.abc
|
||||
|
||||
|
||||
# From PyTorch internals
|
||||
def _ntuple(n):
|
||||
def parse(x):
|
||||
if isinstance(x, container_abcs.Iterable):
|
||||
if isinstance(x, collections.abc.Iterable):
|
||||
return x
|
||||
return tuple(repeat(x, n))
|
||||
return parse
|
||||
|
|
Loading…
Reference in New Issue