Skip to content

backwards

backwards

BackwardsMediator

BackwardsMediator(intervention: Callable, info: 'Tracer.Info', name: Optional[str] = None, batch_group: Optional[List[int]] = None, stop: Optional[int] = None)

Bases: Mediator

request

request(requester: Any)

BackwardsTracer

BackwardsTracer(tensor: Tensor, fn: Callable, *args, **kwargs)

Bases: Invoker

tensor instance-attribute

tensor = tensor

fn instance-attribute

fn = fn

execute

execute(fn: Callable)

wrap_grad

wrap_grad(interleaver: Interleaver)

Create a hook for gradient intervention.

RETURNS DESCRIPTION

A function that can be used to intercept gradients