基础Tensor(张量)操作 TODO 创建线性(玩具)数据集 import torch from torch import nn import matplotlib.pyplot as plt device = "cuda" if torch.cuda.is_available() else "cpu" print(device) # create data weight = 0.6 bias = 0.4 start = 0 end = 1 step = 0.01 X = torch.arange(start, end, step).unsqueeze(dim=1) y = weight * X + bias X[:10], y[:10] train_split = […]