index.rst

pystencils-autodiff

This is the documentation of pystencils-autodiff.

Installation

Install via pip :

pip install pystencils-autodiff

or if you downloaded this repository using:

pip install -e .

Usage

Create a :class:`pystencils.AssignmentCollection` with pystencils:

You can then obtain the corresponding backward assignments:

You can see the derivatives with respective to the two inputs multiplied by the gradient diffz_C of the output z_C.

You can also use the class :class:`.AutoDiffOp` to obtain both the assignments (if you are curious) and auto-differentiable operations for Tensorflow...

... or Torch:

Contents

Indices and tables

• :ref:`genindex`
• :ref:`modindex`
• :ref:`search`