随笔分类 -  深度学习

摘要:一、TensorDataset 语法:class torch.utils.data.TensorDataset(data_tensor, target_tensor) 作用:包装数据和目标张量(类似Python中的zip()函数),可通过第一维度索引两个张量恢复数据。故要保证两个tensor的第一维 阅读全文
posted @ 2023-03-02 17:13 Kruskal 阅读(357) 评论(0) 推荐(0)
摘要:0 梯度更新函式 1 梯度 神经网络参数如下: θ = {w1, w2, ... , b1, b2, ...} 权重梯度如下: 为了更好且有效的计算梯度,我们使用反向传播算法。 2 链式法则 3 反向传播 损失函数(Loss function)是定义在单个训练样本上的,比如我们想要分类,就是预测的类 阅读全文
posted @ 2022-11-30 00:13 Kruskal 阅读(59) 评论(0) 推荐(0)