我有一个二维张量,我想从每一行提取一些起始元素。
每个元素的索引必须小于或等于I行
请注意,不同行的列索引不同。
下面的示例使其合并:
二维张量为:
[[4 2 4 4 1 1 1 1 1 1 1 1 1 1 1 1]
[4 4 4 4 4 4 4 1 1 1 1 1 1 1 1 1]
[4 4 4 5 4 4 4 1 1 1 1 1 1 1 1 1]
[4 4 1 4 4 4 4 1 1 1 1 1 1 1 1 1]
[4 4 4 4 6 4 4 8 8 1 1 1 1 1 1 1]
[3 9 9 9 9 9 9 1 1 1 1 1 1 1 1 1]
[3 9 9 9 9 9 9 1 1 1 1 1 1 1 1 1]
[1 9 9 9 9 9 9 1 1 1 1 1 1 1 1 1]
[3 9 4 9 9 9 9 1 1 1 1 1 1 1 1 1]
[3 9 9 6 9 9 9 1 1 1 1 1 1 1 1 1]]
索引数组是:
^{pr2}$如何从上面的索引数组中获取以下数组:
[[4 2 4 4 ]
[4 4 4 4 4 4 4 ]
[4 4 4 5 4 4 4]
[4 4 1 4 4 4 4]
[4 4 4 4 6 4 4 8 8]
[3 9 9 9 9 9 9 ]
[3 9 9 9 9 9 9 ]
[1 9 9 9 9 9 9 ]
[3 9 4 9 9 9 9 ]
[3 9 9 6 9 9 9 ]]
这里有一种将其作为稀疏张量的方法:
输出:
^{pr2}$它并不是绝对理想的,因为它需要在整个坐标网格下运行。在NumPy中,您可能可以先生成索引,然后从稠密张量中只选择所需的值,但我不确定在TensorFlow中是否可行。在
相关问题 更多 >
编程相关推荐