File tree 1 file changed +7
-5
lines changed
1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -59,8 +59,9 @@ def to_tuple(shape):
59
59
60
60
Returns
61
61
-------
62
- If `shape` is None, returns an empty tuple. If it's an int, (shape,) is
63
- returned. If it is array-like, tuple(shape) is returned.
62
+ shape : tuple
63
+ If `shape` is None, returns an empty tuple. If it's an int, (shape,) is
64
+ returned. If it is array-like, tuple(shape) is returned.
64
65
"""
65
66
if shape is None :
66
67
return tuple ()
@@ -321,9 +322,10 @@ def change_dist_size(
321
322
322
323
Returns
323
324
-------
324
- A new distribution variable that is equivalent to the original distribution with
325
- the new size. The new distribution will not reuse the old RandomState/Generator
326
- input, so it will be independent from the original distribution.
325
+ dist : TensorVariable
326
+ A new distribution variable that is equivalent to the original distribution with
327
+ the new size. The new distribution will not reuse the old RandomState/Generator
328
+ input, so it will be independent from the original distribution.
327
329
328
330
Examples
329
331
--------
You can’t perform that action at this time.
0 commit comments