aeppl.dists.DiracDelta#

class aeppl.dists.DiracDelta(rtol=1e-05, atol=1e-08)[source]#

An Op that represents a Dirac-delta distribution.

Methods

DiracDelta.L_op(inputs, outputs, output_grads)

Construct a graph for the L-operator.

DiracDelta.R_op(inputs, eval_points)

Construct a graph for the R-operator.

DiracDelta.__init__([rtol, atol])

DiracDelta.add_tag_trace([user_line])

Add tag.trace to a node or variable.

DiracDelta.do_constant_folding(fgraph, node)

Determine whether or not constant folding should be performed for the given node.

DiracDelta.get_params(node)

Try to get parameters for the Op when Op.params_type is set to a ParamsType.

DiracDelta.grad(inputs, output_grads)

Construct a graph for the gradient with respect to each input variable.

DiracDelta.infer_shape(fgraph, node, ...)

DiracDelta.make_node(x)

Construct an Apply node that represent the application of this operation to the given inputs.

DiracDelta.make_py_thunk(node, storage_map, ...)

Make a Python thunk.

DiracDelta.make_thunk(node, storage_map, ...)

Create a thunk.

DiracDelta.perform(node, inp, out)

Calculate the function on the inputs and put the variables in the output storage.

DiracDelta.prepare_node(node, storage_map, ...)

Make any special modifications that the Op needs before doing Op.make_thunk.

Attributes

default_output

An int that specifies which output Op.__call__ should return.

destroy_map

A dict that maps output indices to the input indices upon which they operate in-place.

itypes

otypes

params_type

view_map

A dict that maps output indices to the input indices of which they are a view.