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