PyTorch（一）

PyTorch（一） 一、迴歸問題 1、建立數據集 import torch import matplotlib.pyplot as plt x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1) # x data (tensor), shape=(100, 1) y = x.pow(2) + 0.2*torch.rand(x.size()

>>阅读原文<<