例如考虑对象定位问题。这里 NN 将有 5 个输出。output[0]
将告诉图像中存在对象的概率,其他 4 个将告诉边界框坐标。正如我们所看到的,output[0]
它必须使用交叉熵等分类损失,并且output[1]
必须output[4]
使用均方误差等回归损失。
所以总损失是这样的:
loss=Cross_entropy(output[0],Y[0])+MSE(output[1:5],Y[1:5]) #Y is true value
这样的损失是否可以以矢量化形式反向传播?我可以在张量流中实现这种损失吗?如果是,tensorflow 是如何做到的?它是否一次对向量或矩阵的每个元素而不是整个事物执行微分?