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