def gather()

in src/dist_utils.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def gather(x: torch.tensor):
    x_gather = Gather.apply(x)
    x_gather = torch.cat(x_gather, dim=0)
    return x_gather