src/dataset.py

import torch

def get_data(n=200):
  x = torch.linspace(-10,10,n).unsqueeze(1)
  y = 2 * x + 3
  return x, y

if __name__ == '__main__':
    x,y = get_data()
    print(x)
    print(y)