Lines Matching full:autograd

4 Automatic differentiation package - torch.autograd
7 .. automodule:: torch.autograd
8 .. currentmodule:: torch.autograd
49 This section contains the higher level API for the autograd that builds on the basic API above
87 :func:`torch.autograd.backward` or :func:`torch.Tensor.backward`
136 Supporting in-place operations in autograd is a hard matter, and we discourage
137 their use in most cases. Autograd's aggressive buffer freeing and reuse makes
157     use autograd with tensors. Autograd automatically supports Tensors with
173 Tensor autograd functions
244 .. automodule:: torch.autograd.gradcheck
245 .. currentmodule:: torch.autograd.gradcheck
256 .. currentmodule:: torch.autograd
261 Autograd includes a profiler that lets you inspect the cost of different
263 implemented at the moment - CPU-only using :class:`~torch.autograd.profiler.profile`.
265 :class:`~torch.autograd.profiler.emit_nvtx`.
267 :class:`~torch.autograd.profiler.emit_itt`.
269 .. autoclass:: torch.autograd.profiler.profile
288 .. autoclass:: torch.autograd.profiler.emit_nvtx
289 .. autoclass:: torch.autograd.profiler.emit_itt
313 Autograd graph
315 Autograd exposes methods that allow one to inspect the graph and interpose behavior during
318 The ``grad_fn`` attribute of a :class:`torch.Tensor` holds a  :class:`torch.autograd.graph.Node`
319 if the tensor is the output of a operation that was recorded by autograd (i.e., grad_mode is
340     >>> print(isinstance(b.grad_fn, torch.autograd.graph.Node))
353 .. autoclass:: torch.autograd.graph.saved_tensors_hooks
355 .. autoclass:: torch.autograd.graph.save_on_cpu
357 .. autoclass:: torch.autograd.graph.disable_saved_tensors_hooks
359 .. autoclass:: torch.autograd.graph.register_multi_grad_hook
361 .. autoclass:: torch.autograd.graph.allow_mutation_on_saved_tensors
363 .. autoclass:: torch.autograd.graph.GradientEdge
365 .. autofunction:: torch.autograd.graph.get_gradient_edge
371 .. py:module:: torch.autograd.anomaly_mode
372 .. py:module:: torch.autograd.forward_ad
373 .. py:module:: torch.autograd.function
374 .. py:module:: torch.autograd.functional
375 .. py:module:: torch.autograd.grad_mode
376 .. py:module:: torch.autograd.graph
377 .. py:module:: torch.autograd.profiler
378 .. py:module:: torch.autograd.profiler_legacy
379 .. py:module:: torch.autograd.profiler_util
380 .. py:module:: torch.autograd.variable