与えられたテンソル[5,4,3,4]、各行がn個とm個のゼロを持つ定数テンソルを生成する方法、n = 5,4,3,4、およびm = 0,1,2,1 。テンソルA考える

Question

：[5,4,3,4]、私はテンソルBを作成する：

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

Bの各行が有するn個のものはn = 5,4,3,4 Aに従って残りの位置0で埋められます。

これをテンソルフローで実現できますか？

Answer 1

これにはを使用できます。

import tensorflow as tf 

A = tf.constant([5,4,3,4], dtype=tf.int32) 
max_len = tf.reduce_max(A) 
B = tf.sequence_mask(A, max_len, dtype=tf.int32) 
with tf.Session() as sess: 
    print(sess.run(B)) 

プリント：

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