Gram矩阵(pytorch)
风格迁移中广泛使用的
def gram_matrix(y): """ Returns the gram matrix of y (used to compute style loss) """ (b, c, h, w) = y.size() features = y.view(b, c, w * h) features_t = features.transpose(1, 2) #C和w*h转置 gram = features.bmm(features_t) / (c * h * w) #bmm 将features与features_t相乘 return gram
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步