Esempio n. 1
0
def flatten(x):
    return tuple(
        function._iter_filter(
            lambda o: isinstance(o, Variable) or torch.is_tensor(o))(x))
Esempio n. 2
0
def flatten(x):
    return tuple(
        function._iter_filter(lambda o: isinstance(o, torch.Tensor))(x))
Esempio n. 3
0
def flatten(x):
    return tuple(function._iter_filter(lambda o: isinstance(o, torch.Tensor))(x))