Skip to content

Commit 619fec6

Browse files
authored
[Doc] Update discrete.py (#2850)
1 parent 27d3680 commit 619fec6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torchrl/modules/distributions/discrete.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -185,6 +185,7 @@ class MaskedCategorical(D.Categorical):
185185
padding_value: The padding value in the mask tensor. When
186186
sparse_mask == True, the padding_value will be ignored.
187187
188+
Examples:
188189
>>> torch.manual_seed(0)
189190
>>> logits = torch.randn(4) / 100 # almost equal probabilities
190191
>>> mask = torch.tensor([True, False, True, True])

0 commit comments

Comments
 (0)