pytorch
文章平均质量分 87
JacksonKim
这个作者很懒,什么都没留下…
展开
-
torch.Tensor.repeat是什么操作?对比numpy库和torch库中的tile和repeat操作
https://blog.csdn.net/u010496337/article/details/50572866/https://numpy.org/doc/stable/reference/generated/numpy.tile.htmlhttps://numpy.org/doc/stable/reference/generated/numpy.repeat.html原创 2020-12-18 23:38:50 · 1760 阅读 · 1 评论 -
Pytorch三问
1. model.train()有什么用?如果模型中有BN层(Batch Normalization)和Dropout,需要在训练时添加model.train(),在测试时添加model.eval()。其中model.train()是保证BN层用每一批数据的均值和方差,而model.eval()是保证BN用全部训练数据的均值和方差;而对于Dropout,model.train()是随机取一部分网络连接来训练更新参数,而model.eval()是利用到了所有网络连接。参考:资料12. PyTorc原创 2020-11-24 16:02:44 · 244 阅读 · 0 评论 -
Pytorch:torch.Tensor.scatter()和torch.Tensor.gather
一、文档说明1. Tensor.scatter(dim, index, src) → Tensor它是torch.Tensor.scatter_()的错位版本,即:scatter_(dim, index, src, reduce=None) → Tensor该函数用来scatter对于一个3-D的tensor,self会被更新为self[index[i][j][k]][j][k] = src[i][j][k] # if dim == 0self[i][index[i][j][k]][k] =原创 2021-01-20 13:20:45 · 477 阅读 · 1 评论