根据值分割张量

Question

如何为给定张量的N个唯一值生成N个不同的稀疏张量？ 例如，如果我有：

tensor = [[1,3,4,5],[1,2,3,2],[3,3,4,5],[2,2,1,4]]

我希望结果是：

ch1 = [[1,0,0,0],[1,0,0,0],[0,0,0,0],[0,0,1,0]]
ch2 = [[0,0,0,0],[0,1,0,1],[0,0,0,0],[1,1,0,0]]
ch3 = [[0,1,0,0],[0,0,1,0],[0,0,0,0],[0,0,0,0]]
...

我该如何在tensorflow中做到这一点？假设我有一个NHWC格式的张量。

Answer 1

知道了。我们可以使用tf.one_hot()