Tensorflow使用类似于tf.one_hot的0和1初始化列表

时间:2017-06-16 17:33:39

标签: python tensorflow one-hot-encoding

我目前有一个时间序列值列表,表示值为[1, 3, 5, 7, 3]。目前我正在使用tf.one_hot为列表中的每个值获取一个热矢量/张量代表。

1 = [0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0]
3 = [0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0]
5 = [0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0]
7 = [0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0]
3 = [0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0]

在Tensorflow中是否有一种方法/功能可以让我做类似的事情但是将所有值从0初始化为1s的值?

期望的结果:

1 = [1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0]
3 = [1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0]
5 = [1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0]
7 = [1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0]
3 = [1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0]

1 个答案:

答案 0 :(得分:1)

您正在寻找tf.sequence_mask

TensorFlow API