- 博客(5)
- 收藏
- 关注
原创 pytorch实现BP神经网络(精细化手搓版)
可能是全网用《最复杂繁琐的代码》一步步实现一个《由几个全连接构成的最最最基础的BP神经网络》的版本之一。训练过程的损失函数变化:(epoch_num=1000)硬核打工人能手搓就绝不可能掉包!
2023-09-19 16:34:51
742
原创 通过手动计算短平快地搞懂Pytorch中两个函数F.binary_cross_entropy_with_logits及F.cross_entropy的计算原理
例如input是二维的tensor,尺寸是(batch_size,label_num),即batch的长度是batch_size,多分类任务的标签数是label_num,则target是一维的tensor,且类型为LongTensor,每个值代表对应样本真实标签的索引。首先是F.binary_cross_entropy_with_logits(input,target)的手动计算和掉包计算实现。F.cross_entropy(input,target)的手动计算和掉包计算实现。
2023-05-30 10:23:47
623
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人