Function get_batch

regression/main.py:33–38

Builds a batch i.e. (x, f(x)) pair.

(batch_size=32)

def get_batch(batch_size=32):
    """Builds a batch i.e. (x, f(x)) pair."""
    random = torch.randn(batch_size)
    x = make_features(random)
    y = f(x)
    return x, y


# Define model