dbadc63
1
2
3
4
5
6
7
8
9
10
11
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)
