动态切片张量到一个十个形状不均匀的张量列表

2024-04-26 23:10:58 发布

您现在位置:Python中文网/ 问答频道 /正文

给定一个三维源张量tbatch_sizeheightwidth)和另一个表示长度的二维张量l。我想做一些类似的事情:

rslt = tf.dynamic_slice(t, l)

例如rslt[0] = t[:l[0][0], :l[0][1]]rslt[1] = t[:l[1][0], :l[1][1]]等等。但我知道,tf.slice做不到这一点。你能告诉我如何在tensorflow中进行动态切片吗?谢谢!你知道吗


Tags: sizetftensorflowbatch动态切片slicedynamic