将张量（指示位置）转移到蒙版的好方法是什么？

Question

我有一个具有batch_size 2和长度5的loc张量。我想基于loc张量生成mask_trg张量。 位置张量指示当前批次的当前位置。 例如，loc [0，2] = 2表示mask_trg [0，2] = [1、1、1、0、0]，因为当前位置是2，所有路径都标记为1。

我尝试了多种方法，但不确定哪种有效的方法（基于矢量）来生成mask_trg张量？

loc = tf.constant(
    [
        [1, 1, 2, 2, 4],
        [1, 2, 4, 4, 4]
    ])

mask_trg = tf.constant(
    [
        [
            [1, 1, 0, 0, 0],
            [1, 1, 0, 0, 0],
            [1, 1, 1, 0, 0],
            [1, 1, 1, 0, 0],
            [1, 1, 1, 1, 1]
        ],
        [
            [1, 1, 0, 0, 0],
            [1, 1, 1, 0, 0],
            [1, 1, 1, 1, 1],
            [1, 1, 1, 1, 1],
            [1, 1, 1, 1, 1]
        ],
    ])

Answer 1

NaN

我认为y2是您想要的。