我有一个大小为（12,12）的2d张量G。张量的每个单元都是一个浮点。 我还有另一个2d张量M（12,5）。
我将创建一个大小为（12、5）的张量sub_G。示例如下：
M = [ [1 2 4 7 5 ]
[1 5 8 9 11]
[6 2 8 10 5]
...
]
Expected Result:
sub_G_row [0] = [ G[0, M[0][0]], G[0, M[0][1]] , G[0, M[0,2]] , G[0, M[0,3]] , G[0, M[0,4]] ]
//[ G[0,1] , G[0,2], G[0,4], G[0,7], G[0,5]
sub_G_row [1] = [G[0, M[1][0]], G[1, M[1][1]] , G[1, M[1,2]] , G[1, M[1,3]] , G[1, M[1,4]] ]
//[ G[0,1] , G[0,5], G[0,8], G[0,9], G[0,11]
...
sub_G_row[i] = [ G[i, M[i][0]], G[i, M[i][1]] , G[i, M[i,2]] , G[i, M[i,3]] , G[i, M[i,4]] ]
有人可以帮我吗？
我曾经尝试过使用collect_nd等，但是我是tensorflow的新手。