张量流是否支持非标准化损失?

时间:2018-01-23 17:07:19

标签: tensorflow

我正在使用张量流框架进行语义分割。我的问题是,张量流是否支持具有交叉熵的softmax的非标准化损失,例如如下:

https://github.com/torrvision/caffe/blob/d779ee369bf86739bc225d99e78aefa040da199f/src/caffe/layers/softmax_loss_layer.cpp#L115

1 个答案:

答案 0 :(得分:0)

很难跟踪该文件中发生的事情。我相信“非标准化”是指softmax的输出产生[0,1]范围内的值,不需要求和为1,例如多类分类所使用的值。

如果我是对的,你只是在寻找在输出层上应用sigmoid的tf.nn.sigmoid_cross_entropy_with_logits(...)。每个输出神经元产生[0,1]范围内的值,但它们彼此独立。

这是您应该用于多类分类的。