- 博客(3)
- 收藏
- 关注
原创 class torch.nn.MarginRankingLoss(margin=0, size_average=True)
功能: 计算两个向量之间的相似度,当两个向量之间的距离大于margin,则loss为正,小于margin,loss为0。y == 1时,x1要比x2大,才不会有loss,反之,y == -1 时,x1要比x2小,才不会有loss。
2020-11-24 15:38:18 3584
原创 自适应池化
# target output size of 5x7>>> m = nn.AdaptiveAvgPool2d((5,7))>>> input = torch.randn(1, 64, 8, 9)>>> output = m(input)>>> # target output size of 7x7 (square)>>> m = nn.AdaptiveAvgPool2d(7)>>> i.
2020-11-24 14:42:08 888
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人