pytorch 基于masking对元素进行替换

描述

pytorch 基于masking对元素进行替换. 代码如下. 先展平再赋值.

代码

# map.shape [64,60,128]
# infill.shape [64,17,128]
# mask_indices.shape [64,60]
   map = map.reshape(
            map.shape[0] * map.shape[1],
            map.shape[2]) [mask_indices.reshape(mask_indices.shape[0]*mask_indices.shape[1])] \
            = fillin.reshape(fillin.shape[0]*fillin.shape[1], fillin.shape[2])