def chunk_tensor()

in dualpipe/utils.py [0:0]

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

def chunk_tensor(x, chunks, dim):
    if x is None:
        return [None for _ in range(chunks)]
    return x.tensor_split(chunks, dim=dim)