Created by: luluxing3

有一个输入tensor，shape= [batch, seq_len, dim] 和记录position的tensor，shape=[batch];按照position取具体seq_len的某一个时刻(切片[batch，1，dim]),最终得到输出tensor，shape=[batch, dim] 请问如何在paddle中实现？