摘要:
loss = torch.mean((z_q.detach()-z)**2) + self.beta * torch.mean((z_q - z.detach()) ** 2) z_q是codebook 找到的最接近z的向量. z是encoder生成的向量. L对z求导 = 2(z_q.detach 阅读全文
posted @ 2023-11-24 17:28
张博的博客
阅读(571)
评论(0)
推荐(0)
contact me:wechat 15122306087