【PyTorch深度学习-龙龙老师】-测试版202112PyTorch 中常见的数学运算函数。 4.9.1 加、减、乘、除运算 加、减、乘、除是最基本的数学运算,分别通过 torch.add、torch.sub、torch.mul、 torch.div 函数实现,PyTorch 已经重载了+、 − 、 ∗ 、/运算符,通常推荐直接使用运算符 来完成加、减、乘、除运算,简单清晰。 预览版202112 第 4 章 PyTorch 基础 34 -gaussians kl_div = -0.5 * (log_var + 1 - mu**2 - torch.exp(log_var)) kl_div = kl_div.mean() # 合并 Loss 项目 loss = rec_loss + 10. * kl_div # 自动求导,同时更新 encoder0 码力 | 439 页 | 29.91 MB | 1 年前3
共 1 条
- 1
 













