如何在 Tensorflow 中基于索引集获取张量的子集

问题描述 投票:0回答:0

给出 a: 张量([[3., 5.], [1., 2.],[5., 7.],[4., 6.],[3., 5.]]) 指数:张量([0,0],[3,1])

有什么方法可以得到 [index] 是 [3,6]

tensorflow tensor
© www.soinside.com 2019 - 2024. All rights reserved.