假设我有一个大小为BxWxHxD的张量.我要处理的张量,使得我有一个新的BxWxHxD张量仅在每个宽x高片的最大元素保持,而所有其他值都为零.
换句话说,我认为实现这一目标的最佳方法是以某种方式在WxH切片上采用2D argmax,从而产生行和列的BxD索引张量,然后可以将其转换为单热BxWxHxD张量以用作一张面具.我该如何工作？
tensorflow
tensorflow ×1