def slice_fields()

in shap_e/models/stf/renderer.py [0:0]

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

def slice_fields(fields: torch.Tensor, dim: int, start: int, end: int):
    if dim == 1:
        return fields[:, start:end]
    elif dim == 2:
        return fields[:, :, start:end]
    elif dim == 3:
        return fields[:, :, :, start:end]
    else:
        raise ValueError(f"cannot slice dimension {dim}")