Function step_function

benchmarks/python/comparative/bench_torch.py:298–302

(x)

@torch.no_grad()
def step_function(x):
    y = x
    for i in range(100):
        y = torch.where(y < 0, 0, 1)
    sync_if_needed(x)


@torch.no_grad()