tensorflow怎么获取最后一个维度特定索引的张量?

Question

我有一个TensorFlow张量[1,513,513,21],其中的513,513分别表示模型预测的特征图的宽度和高度，21表示每个像素预测的21个类别的置信度，之前用tf.argmax返回的是最后一个维度，也就是21这个维度上最大那个置信度的值，给到每个像素点对应。可我现在想要输出每一个类别的，比如说每个像素对应给出0类别识别那个值，这样就有21个特征图出来，而不是一张。

predictions[output] = tf.argmax(logits, 3)

Answer 1

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 这有个类似的问题, 你可以参考下: https://ask.csdn.net/questions/7666992
• 我还给你找了一篇非常好的博客，你可以看看是否有帮助，链接：记录使用tensorflow时的一些报错
• 你还可以看下tensorflow参考手册中的 tensorflow 张量

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^