首页 > 解决方案 > 如何在tensorflow中获取第一批的损失张量

问题描述

为了实现本文[https://arxiv.org/abs/1711.02257]中的grad norm方法,我需要计算loss_i/loss_0,其中loss_i是第i步的损失。

现在的问题是如何将loss_0存储到张量中?

标签: tensorflow

解决方案


推荐阅读