import torch


def random_data(N, D_in, D_out):
    """
    Generates random data for training/testing the PyTorch model.

    N is the data size
    D_in is the input dimension
    D_out is the output dimension
    """

    # Create random Tensors to hold inputs and outputs
    x = torch.randn(N, D_in)
    y = torch.randn(N, D_out)
    return {'X_train': x, 'Y_train': y, 'X_test': x, 'Y_test': y}