def build_from_tensor_shapes()

in dualpipe/comm.py [0:0]

• 2 lines of code
• 2 McCabe index (conditional complexity)

def build_from_tensor_shapes():
    return [torch.empty(s, dtype=TENSOR_DTYPE, device="cuda", requires_grad=True) for s in TENSOR_SHAPES]