Question

for i in range shape(L)[0]:
    L[i,:,:,:,0] = i
for j in range shape(L)[2]:
    L[:,:,j,:,1] = j

例如，如果我想实现上面的代码，该怎么办：

（实际上，我倾向于使用tf.gather，因此我需要创建一个索引）

Answer 1

L的类型是什么？如果它是一个张量，你将不得不通过一个tf.py_func函数（我认为这是最好的选择，因为它可以让你把你的张量对象当作一个numpy.array而且从那时起它＆＃39 ; s trivial）cf：

其他看起来很复杂cf：