(self, pub_obses, range_idxs, legal_action_masks)
return out
Softmax is not applied in here! It is separate in training and action fns
Softmax is not applied in here! It is separate in training and action fns
Softmax is not applied in here! It is separate in training and action fns
def forward(self, pub_obses, range_idxs, legal_action_masks):
        """
        Softmax is not applied in here! It is separate in training and action fns
        """

        out = self._mpm(pub_obses=pub_obses, range_idxs=range_idxs)
        out = self._relu(self._final_layer(out))
        out = self._out_layer(out)
        out = torch.where(legal_action_masks == 1,
                          out,
                          torch.FloatTensor([-10e20]).to(device=out.device).expand_as(out))
        return out
def forward ( self , pub_obses , range_idxs , legal_action_masks ) : out = self . _mpm ( pub_obses = pub_obses , range_idxs = range_idxs ) out = self . _relu ( self . _final_layer ( out ) ) out = self . _out_layer ( out ) out = torch . where ( legal_action_masks == 1 , out , torch . FloatTensor ( [ - 10e20 ] ) . to ( device = out . device ) . expand_as ( out ) ) return out