如何将字符串Tensor转换为Tensor浮点列表?

时间:2018-02-24 16:00:28

标签: tensorflow casting shape reshape tensor

所以,我的代码就像

    parsed_line = tf.decode_csv(line, [[0], [0], [""]])
    print(parsed_line[0])
    del parsed_line[0]
    del parsed_line[0]
    features = parsed_line
    print(parsed_line[0])

然后结果是

    [<tf.Tensor 'DecodeCSV:0' shape=() dtype=int32>, <tf.Tensor 'DecodeCSV:1' shape=() dtype=int32>, <tf.Tensor 'DecodeCSV:2' shape=() dtype=string>]

    [<Tensor("DecodeCSV:2", shape=(), dtype=string)>]

我将给这个解码函数的csv是

    1, 0, 0101010010101010101010

我希望这个&#34; 0101010010101010101010&#34;到

    [0,1,0,1,0,.........] 
张量流中的

    [<Tensor("DecodeCSV:2", shape=(), dtype=string)>]

    [<tf.Tensor 'DecodeCSV:0' shape=() dtype=int32>, <tf.Tensor 'DecodeCSV:1' shape=() dtype=int32>, ............]
你有什么想法吗?

1 个答案:

答案 0 :(得分:3)

您可以使用tf.string_splittf.string_to_number

这样做
import tensorflow as tf

line = tf.constant("1000111101", shape=(1,))
b = tf.string_split(line, delimiter="").values
c = tf.string_to_number(b, tf.int32)
print(c)
with tf.Session() as sess:
    print(sess.run(c))
  

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