feature = ... (batch_size, month, x,y, channel)
attention_weight = ... (batch_size, month, x,y,1) where attention_weight.sum(1) = 1
selected feature = (feature*attention_weight).sum(1) = ... (batch_size,x,y, channel)
predicted_agbm = decode ... (batch_size,x,y, channel) = ... (batch_size,x,y, 1)
3 Likes