给定一个 3D 源张量t
( batch_size
, height
, ) 和另一个指示长度的width
2D 张量。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 中进行动态切片吗?非常感谢!
给定一个 3D 源张量t
( batch_size
, height
, ) 和另一个指示长度的width
2D 张量。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 中进行动态切片吗?非常感谢!