inverse get_mask

agentnet/learning/generic.py

@@ -291,3 +291,13 @@ def get_end_indicator(is_alive, force_end_at_t_max=False):
 
     return is_end
 
+def get_mask_by_eos(is_eos):
+    """takes indicator of "it ends now", returns mask. Ignores everything after first end.
+    :param is_eos: indicator that is 0 for all
+    :type is_eos: theano.matrix
+    """
+    assert is_eos.ndim==2
+    is_right_after_eos = T.concatenate([T.zeros_like(is_eos[:,:1]),is_eos[:,:-1]],-1)
+    is_after_eos = T.eq(T.cumsum(is_right_after_eos,axis=-1),0).astype('uint8')
+    return is_after_eos
+