We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 27d3680 commit 619fec6Copy full SHA for 619fec6
torchrl/modules/distributions/discrete.py
@@ -185,6 +185,7 @@ class MaskedCategorical(D.Categorical):
185
padding_value: The padding value in the mask tensor. When
186
sparse_mask == True, the padding_value will be ignored.
187
188
+ Examples:
189
>>> torch.manual_seed(0)
190
>>> logits = torch.randn(4) / 100 # almost equal probabilities
191
>>> mask = torch.tensor([True, False, True, True])
0 commit comments