- 博客(3)
- 收藏
- 关注
RSS订阅原创 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
3543
原创 自适应池化
# 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
861
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人