Revert "replace reduce to reduction, better error message for invalid mask" (#3182)

Author: vmoens

torchrl/modules/distributions/discrete.py

@@ -251,10 +251,7 @@ def __init__(
             sparse_mask = True
         else:
             sparse_mask = False
-        if !mask.any():
-            raise ValueError(
-                f"Provided ``mask`` must contain a value True"
-            )
+
         if probs is not None:
             if logits is not None:
                 raise ValueError(
@@ -363,7 +360,7 @@ def log_prob(self, value: torch.Tensor) -> torch.Tensor:
                         original_value_shape = value.shape
                         value = value.flatten()
                     logits = logits.unsqueeze(0).expand(value.shape + logits.shape)
-                result = -torch.nn.functional.cross_entropy(logits, value, reduction='none')
+                result = -torch.nn.functional.cross_entropy(logits, value, reduce=False)
                 if original_value_shape is not None:
                     result = result.unflatten(0, original_value_shape)
             else:
@@ -394,7 +391,7 @@ def log_prob(self, value: torch.Tensor) -> torch.Tensor:
                     original_idx_shape = idx.shape
                     idx = idx.flatten()
                 logits = logits.unsqueeze(0).expand(idx.shape + logits.shape)
-            ret = -torch.nn.functional.cross_entropy(logits, idx, reduction='none')
+            ret = -torch.nn.functional.cross_entropy(logits, idx, reduce=False)
             if original_idx_shape is not None:
                 ret = ret.unflatten(0, original_idx_shape)
         else: