eazygrad.no_grad_ctx¶

class eazygrad.no_grad_ctx[source]¶

Bases: object

Context manager that temporarily disables gradient tracking.

Examples

>>> with eazygrad.no_grad_ctx():
...     y = x + 1.0