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