Skip to content

Commit a33dfdb

Browse files
committed
Rename variables in choice_gradients docstring
1 parent a3b57de commit a33dfdb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/gen_fn_interface.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -373,7 +373,7 @@ function accumulate_param_gradients!(trace)
373373
end
374374

375375
"""
376-
(arg_grads, choice_values, choice_gradient) = choice_gradients(
376+
(arg_grads, value_choices, gradient_choices) = choice_gradients(
377377
trace, selection=EmptySelection(), retgrad=nothing)
378378
379379
Given a previous trace \$(x, t)\$ (`trace`) and a gradient with respect to the
@@ -389,15 +389,15 @@ If an argument is not annotated with `(grad)`, the corresponding value in
389389
`arg_grads` will be `nothing`.
390390
391391
Also given a set of addresses \$A\$ (`selection`) that are continuous-valued
392-
random choices, return the folowing gradient (`choice_grads`) with respect to
392+
random choices, return the folowing gradient (`gradient_choices`) with respect to
393393
the values of these choices:
394394
```math
395395
∇_A \\left( \\log P(t; x) + J \\right)
396396
```
397397
The gradient is represented as a choicemap whose value at (hierarchical)
398398
address `addr` is \$∂J/∂t[\\texttt{addr}]\$.
399399
400-
Also return the choicemap (`choice_values`) that is the restriction of \$t\$ to \$A\$.
400+
Also return the choicemap (`value_choices`) that is the restriction of \$t\$ to \$A\$.
401401
"""
402402
function choice_gradients(trace, selection::Selection, retgrad)
403403
error("Not implemented")

0 commit comments

Comments
 (0)