Created by: DZhaoXd

现遇到一个坐标索引问题，tensor A 的 shape 是 [b, w, h]，b是batchsize， w,h分别是长和宽；tensor B 的 shape 是 [b, 3, 2]，表示对batch中的每一个数据都存在3个需要索引的坐标（x, y）。请问怎么根据tensor B 去索引 tensor A得到一个[b, 3]的结果？