gpt4 book ai didi

python - 通过重复最后一个元素填充张量(tensorflow)

转载 作者:太空宇宙 更新时间:2023-11-04 02:52:38 25 4
gpt4 key购买 nike

我有以下形状的视频帧张量 (frames, 128, 128, 3) 其中帧在 25 到 36 之间变化。现在我希望张量具有固定大小 (36, 128, 128, 3) 通过将最后一帧重复 36 帧 次。

我可以通过以下方式提取形状为 (128, 128, 3) 的最后一帧:

shape = tf.shape(视频)
last_frame = tf.gather_nd(video, [shape[0]-1])

我现在想重复 36-shape[0] 次并将其添加到 video 的末尾。我该怎么做?

谢谢。

最佳答案

您可以通过选择最后一个元素,使用 tf.tile 将其复制 n 次,然后连接您的结果来做到这一点。

last = frames[-1]
last_35 = tf.tile(tf.expand_dims(last, 0), [35-tf.shape(frames)[0] ,1,1,1])
frames = tf.concat([frames, last_35], 0)

关于python - 通过重复最后一个元素填充张量(tensorflow),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43372273/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com