Pytorch 1.2, tacotron gst model.
I wonna try to mask out gst outputs.
I have two 3d array with shape [32, 177, 512]
                    
            
           1


                    
            
          


So i just wonna have zeros in first tensor like in second
Taking mask and expand shape
gst_mask = ~get_mask_from_lengths(text_lengths)
gst_mask = gst_mask.unsqueeze(-1).expand_as(gst_outputs)
Seems ok
                    
            
          


Then i do
gst_outputs.data.masked_fill_(gst_mask, 0.0)
But get unexpected output: