File tree Expand file tree Collapse file tree 1 file changed +15
-0
lines changed
Expand file tree Collapse file tree 1 file changed +15
-0
lines changed Original file line number Diff line number Diff line change @@ -90,18 +90,33 @@ def set_default_tensor_type(t):
9090
9191
9292def set_rng_state (new_state ):
93+ r"""Sets the random number generator state.
94+
95+ Args:
96+ new_state (torch.ByteTensor): The desired state
97+ """
9398 default_generator .set_state (new_state )
9499
95100
96101def get_rng_state ():
102+ r"""Returns the random number generator state as a ByteTensor."""
97103 return default_generator .get_state ()
98104
99105
100106def manual_seed (seed ):
107+ r"""Sets the seed for generating random numbers. And returns a
108+ `torch._C.Generator` object.
109+
110+ Args:
111+ seed (int or long): The desired seed.
112+ """
101113 return default_generator .manual_seed (seed )
102114
103115
104116def initial_seed ():
117+ r"""Returns the initial seed for generating random numbers as a
118+ python `long`.
119+ """
105120 return default_generator .initial_seed ()
106121
107122
You can’t perform that action at this time.
0 commit comments