tf.map_fn() can have many inputs / outputs. You could thus use tf.range() to build a tensor of row indices and use it along:
確定! 回上一頁