2302_78201170的博客

原创 mylinear代码解析

定义了一个函数create_data,用于生成数据集x是输入特征，通过torch.normal生成均值为0，标准差为1的正态分布随机数，形状为（data_num,len(w),即data_num个样本，每个样本有len(w)个特征y是目标值，通过矩阵相乘torch.matmul（x，w）计算得到，再加上偏置b为了模拟真实数据中的噪声，有生成了一个均值为0，标准差为0.01的正态分布噪声noise，并将其加到y上。