tensorflow - 如何在 tensorflow 中创建掩模张量？

Question

为给定索引创建一定深度的张量，值为 1

索引张量:

[[1 3]
 [2 4]
 [0 4]]

输出张量(深度=5):

[[0. 1. 0. 1. 0.]
 [0. 0. 1. 0. 1.]
 [1. 0. 0. 0. 1.]]

Answer 1

您可以通过首先转换为完整索引，然后使用 sparse_to_dense 函数将这些索引值设置为 1 来实现上述目的。

#Get full indices
mesh = tf.meshgrid(tf.range(indices.shape[1]), tf.range(indices.shape[0]))[1]
full_indices = tf.reshape(tf.stack([mesh, indices], axis=2), [-1,2])

#Output
[[0 1]
 [0 3]
 [1 2]
 [1 4]
 [2 0]
 [2 4]]

#use the above indices and set the output to 1.
#depth_x = 3, depth_y = 5
dense = tf.sparse_to_dense(full_indices,tf.constant([depth_x,depth_y]), tf.ones(tf.shape(full_indices)[0]))

# Output
#[[0. 1. 0. 1. 0.]
#[0. 0. 1. 0. 1.]
#[1. 0. 0. 0. 1.]]