我有一个A尺寸的3D矩阵h x w x c.我想ph x pw从每个"频道"中提取尺寸的片段c.ph划分h和pw分裂w.在这个例子中,
A
h x w x c
ph x pw
c
ph
h
pw
w
h x w x c = 4 x 4 x 3 ph x pw = 2 x 2
我知道如何在tensorflow中使用这个,gather_nd但是我希望在设置它时能有更高效的东西,因为维度很大而且我宁愿没有gather_nd内存中的indices数组.可能有一个聪明的重塑?numpy或tensorflow解决方案都非常好!
gather_nd