Skip to content

V0.4.1 docs #43

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Jul 26, 2018
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
File renamed without changes.
Binary file added docs/0.4.0/_images/ELU.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Hardshrink.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Hardtanh.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/LeakyReLU.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/LogSigmoid.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/PReLU.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/ReLU.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/ReLU6.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/SELU.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Sigmoid.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Softplus.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Softshrink.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Softsign.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Tanh.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added docs/0.4.0/_images/Tanhshrink.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
909 changes: 909 additions & 0 deletions docs/0.4.0/_modules/index.html

Large diffs are not rendered by default.

1,087 changes: 1,087 additions & 0 deletions docs/0.4.0/_modules/torch.html

Large diffs are not rendered by default.

1,019 changes: 1,019 additions & 0 deletions docs/0.4.0/_modules/torch/_tensor_str.html

Large diffs are not rendered by default.

1,057 changes: 1,057 additions & 0 deletions docs/0.4.0/_modules/torch/_utils.html

Large diffs are not rendered by default.

967 changes: 967 additions & 0 deletions docs/0.4.0/_modules/torch/autograd.html

Large diffs are not rendered by default.

1,168 changes: 1,168 additions & 0 deletions docs/0.4.0/_modules/torch/autograd/function.html

Large diffs are not rendered by default.

902 changes: 902 additions & 0 deletions docs/0.4.0/_modules/torch/autograd/grad_mode.html

Large diffs are not rendered by default.

1,375 changes: 1,375 additions & 0 deletions docs/0.4.0/_modules/torch/autograd/profiler.html

Large diffs are not rendered by default.

1,349 changes: 1,349 additions & 0 deletions docs/0.4.0/_modules/torch/cuda.html

Large diffs are not rendered by default.

1,001 changes: 1,001 additions & 0 deletions docs/0.4.0/_modules/torch/cuda/comm.html

Large diffs are not rendered by default.

873 changes: 873 additions & 0 deletions docs/0.4.0/_modules/torch/cuda/nvtx.html

Large diffs are not rendered by default.

914 changes: 914 additions & 0 deletions docs/0.4.0/_modules/torch/cuda/random.html

Large diffs are not rendered by default.

1,007 changes: 1,007 additions & 0 deletions docs/0.4.0/_modules/torch/cuda/streams.html

Large diffs are not rendered by default.

1,349 changes: 1,349 additions & 0 deletions docs/0.4.0/_modules/torch/distributed.html

Large diffs are not rendered by default.

894 changes: 894 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/bernoulli.html

Large diffs are not rendered by default.

882 changes: 882 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/beta.html

Large diffs are not rendered by default.

901 changes: 901 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/binomial.html

Large diffs are not rendered by default.

908 changes: 908 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/categorical.html

Large diffs are not rendered by default.

864 changes: 864 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/cauchy.html

Large diffs are not rendered by default.

823 changes: 823 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/chi2.html

Large diffs are not rendered by default.

1,004 changes: 1,004 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/constraint_registry.html

Large diffs are not rendered by default.

1,045 changes: 1,045 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/constraints.html

Large diffs are not rendered by default.

895 changes: 895 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/dirichlet.html

Large diffs are not rendered by default.

1,020 changes: 1,020 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/distribution.html

Large diffs are not rendered by default.

857 changes: 857 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/exp_family.html

Large diffs are not rendered by default.

868 changes: 868 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/exponential.html

Large diffs are not rendered by default.

868 changes: 868 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/fishersnedecor.html

Large diffs are not rendered by default.

871 changes: 871 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/gamma.html

Large diffs are not rendered by default.

874 changes: 874 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/geometric.html

Large diffs are not rendered by default.

853 changes: 853 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/gumbel.html

Large diffs are not rendered by default.

884 changes: 884 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/independent.html

Large diffs are not rendered by default.

1,434 changes: 1,434 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/kl.html

Large diffs are not rendered by default.

867 changes: 867 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/laplace.html

Large diffs are not rendered by default.

846 changes: 846 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/log_normal.html

Large diffs are not rendered by default.

898 changes: 898 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/multinomial.html

Large diffs are not rendered by default.

988 changes: 988 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/multivariate_normal.html

Large diffs are not rendered by default.

884 changes: 884 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/normal.html

Large diffs are not rendered by default.

885 changes: 885 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/one_hot_categorical.html

Large diffs are not rendered by default.

849 changes: 849 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/pareto.html

Large diffs are not rendered by default.

857 changes: 857 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/poisson.html

Large diffs are not rendered by default.

913 changes: 913 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/relaxed_bernoulli.html

Large diffs are not rendered by default.

911 changes: 911 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/relaxed_categorical.html

Large diffs are not rendered by default.

874 changes: 874 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/studentT.html

Large diffs are not rendered by default.

922 changes: 922 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/transformed_distribution.html

Large diffs are not rendered by default.

1,328 changes: 1,328 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/transforms.html

Large diffs are not rendered by default.

879 changes: 879 additions & 0 deletions docs/0.4.0/_modules/torch/distributions/uniform.html

Large diffs are not rendered by default.

1,222 changes: 1,222 additions & 0 deletions docs/0.4.0/_modules/torch/functional.html

Large diffs are not rendered by default.

863 changes: 863 additions & 0 deletions docs/0.4.0/_modules/torch/multiprocessing.html

Large diffs are not rendered by default.

2,859 changes: 2,859 additions & 0 deletions docs/0.4.0/_modules/torch/nn/functional.html

Large diffs are not rendered by default.

1,204 changes: 1,204 additions & 0 deletions docs/0.4.0/_modules/torch/nn/init.html

Large diffs are not rendered by default.

1,582 changes: 1,582 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/activation.html

Large diffs are not rendered by default.

1,060 changes: 1,060 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/batchnorm.html

Large diffs are not rendered by default.

1,074 changes: 1,074 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/container.html

Large diffs are not rendered by default.

1,618 changes: 1,618 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/conv.html

Large diffs are not rendered by default.

867 changes: 867 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/distance.html

Large diffs are not rendered by default.

978 changes: 978 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/dropout.html

Large diffs are not rendered by default.

1,038 changes: 1,038 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/instancenorm.html

Large diffs are not rendered by default.

918 changes: 918 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/linear.html

Large diffs are not rendered by default.

1,788 changes: 1,788 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/loss.html

Large diffs are not rendered by default.

1,752 changes: 1,752 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/module.html

Large diffs are not rendered by default.

1,020 changes: 1,020 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/normalization.html

Large diffs are not rendered by default.

1,276 changes: 1,276 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/padding.html

Large diffs are not rendered by default.

839 changes: 839 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/pixelshuffle.html

Large diffs are not rendered by default.

1,776 changes: 1,776 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/pooling.html

Large diffs are not rendered by default.

1,560 changes: 1,560 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/rnn.html

Large diffs are not rendered by default.

1,043 changes: 1,043 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/sparse.html

Large diffs are not rendered by default.

1,018 changes: 1,018 additions & 0 deletions docs/0.4.0/_modules/torch/nn/modules/upsampling.html

Large diffs are not rendered by default.

956 changes: 956 additions & 0 deletions docs/0.4.0/_modules/torch/nn/parallel/data_parallel.html

Large diffs are not rendered by default.

1,272 changes: 1,272 additions & 0 deletions docs/0.4.0/_modules/torch/nn/parallel/distributed.html

Large diffs are not rendered by default.

823 changes: 823 additions & 0 deletions docs/0.4.0/_modules/torch/nn/parameter.html

Large diffs are not rendered by default.

859 changes: 859 additions & 0 deletions docs/0.4.0/_modules/torch/nn/utils/clip_grad.html

Large diffs are not rendered by default.

1,130 changes: 1,130 additions & 0 deletions docs/0.4.0/_modules/torch/nn/utils/rnn.html

Large diffs are not rendered by default.

917 changes: 917 additions & 0 deletions docs/0.4.0/_modules/torch/nn/utils/weight_norm.html

Large diffs are not rendered by default.

954 changes: 954 additions & 0 deletions docs/0.4.0/_modules/torch/onnx.html

Large diffs are not rendered by default.

874 changes: 874 additions & 0 deletions docs/0.4.0/_modules/torch/optim/adadelta.html

Large diffs are not rendered by default.

892 changes: 892 additions & 0 deletions docs/0.4.0/_modules/torch/optim/adagrad.html

Large diffs are not rendered by default.

904 changes: 904 additions & 0 deletions docs/0.4.0/_modules/torch/optim/adam.html

Large diffs are not rendered by default.

884 changes: 884 additions & 0 deletions docs/0.4.0/_modules/torch/optim/adamax.html

Large diffs are not rendered by default.

880 changes: 880 additions & 0 deletions docs/0.4.0/_modules/torch/optim/asgd.html

Large diffs are not rendered by default.

1,047 changes: 1,047 additions & 0 deletions docs/0.4.0/_modules/torch/optim/lbfgs.html

Large diffs are not rendered by default.

1,172 changes: 1,172 additions & 0 deletions docs/0.4.0/_modules/torch/optim/lr_scheduler.html

Large diffs are not rendered by default.

1,007 changes: 1,007 additions & 0 deletions docs/0.4.0/_modules/torch/optim/optimizer.html

Large diffs are not rendered by default.

898 changes: 898 additions & 0 deletions docs/0.4.0/_modules/torch/optim/rmsprop.html

Large diffs are not rendered by default.

875 changes: 875 additions & 0 deletions docs/0.4.0/_modules/torch/optim/rprop.html

Large diffs are not rendered by default.

905 changes: 905 additions & 0 deletions docs/0.4.0/_modules/torch/optim/sgd.html

Large diffs are not rendered by default.

900 changes: 900 additions & 0 deletions docs/0.4.0/_modules/torch/optim/sparse_adam.html

Large diffs are not rendered by default.

907 changes: 907 additions & 0 deletions docs/0.4.0/_modules/torch/random.html

Large diffs are not rendered by default.

1,275 changes: 1,275 additions & 0 deletions docs/0.4.0/_modules/torch/serialization.html

Large diffs are not rendered by default.

797 changes: 797 additions & 0 deletions docs/0.4.0/_modules/torch/sparse.html

Large diffs are not rendered by default.

916 changes: 916 additions & 0 deletions docs/0.4.0/_modules/torch/storage.html

Large diffs are not rendered by default.

1,184 changes: 1,184 additions & 0 deletions docs/0.4.0/_modules/torch/tensor.html

Large diffs are not rendered by default.

945 changes: 945 additions & 0 deletions docs/0.4.0/_modules/torch/utils/checkpoint.html

Large diffs are not rendered by default.

1,526 changes: 1,526 additions & 0 deletions docs/0.4.0/_modules/torch/utils/cpp_extension.html

Large diffs are not rendered by default.

1,250 changes: 1,250 additions & 0 deletions docs/0.4.0/_modules/torch/utils/data/dataloader.html

Large diffs are not rendered by default.

911 changes: 911 additions & 0 deletions docs/0.4.0/_modules/torch/utils/data/dataset.html

Large diffs are not rendered by default.

854 changes: 854 additions & 0 deletions docs/0.4.0/_modules/torch/utils/data/distributed.html

Large diffs are not rendered by default.

946 changes: 946 additions & 0 deletions docs/0.4.0/_modules/torch/utils/data/sampler.html

Large diffs are not rendered by default.

1,002 changes: 1,002 additions & 0 deletions docs/0.4.0/_modules/torch/utils/ffi.html

Large diffs are not rendered by default.

925 changes: 925 additions & 0 deletions docs/0.4.0/_modules/torch/utils/model_zoo.html

Large diffs are not rendered by default.

91 changes: 91 additions & 0 deletions docs/0.4.0/_sources/autograd.rst.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,91 @@
.. role:: hidden
:class: hidden-section

Automatic differentiation package - torch.autograd
==================================================

.. automodule:: torch.autograd
.. currentmodule:: torch.autograd

.. autofunction:: backward

.. autofunction:: grad

.. _locally-disable-grad:

Locally disabling gradient computation
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: no_grad

.. autoclass:: enable_grad

.. autoclass:: set_grad_enabled

In-place operations on Tensors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

Supporting in-place operations in autograd is a hard matter, and we discourage
their use in most cases. Autograd's aggressive buffer freeing and reuse makes
it very efficient and there are very few occasions when in-place operations
actually lower memory usage by any significant amount. Unless you're operating
under heavy memory pressure, you might never need to use them.

In-place correctness checks
---------------------------

All :class:`Tensor` s keep track of in-place operations applied to them, and
if the implementation detects that a tensor was saved for backward in one of
the functions, but it was modified in-place afterwards, an error will be raised
once backward pass is started. This ensures that if you're using in-place
functions and not seeing any errors, you can be sure that the computed
gradients are correct.

Variable (deprecated)
^^^^^^^^^^^^^^^^^^^^^

.. warning::
The Variable API has been deprecated: Variables are no longer necessary to
use autograd with tensors. Autograd automatically supports Tensors with
``requires_grad`` set to ``True``. Below please find a quick guide on what
has changed:

- ``Variable(tensor)`` and ``Variable(tensor, requires_grad)`` still work as expected,
but they return Tensors instead of Variables.
- ``var.data`` is the same thing as ``tensor.data``.
- Methods such as ``var.backward(), var.detach(), var.register_hook()`` now work on tensors
with the same method names.

In addition, one can now create tensors with ``requires_grad=True`` using factory
methods such as :func:`torch.randn`, :func:`torch.zeros`, :func:`torch.ones`, and others
like the following:

``autograd_tensor = torch.randn((2, 3, 4), requires_grad=True)``

Tensor autograd functions
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.. autoclass:: torch.Tensor
:members: backward, detach, detach_, register_hook, retain_grad

:hidden:`Function`
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

.. autoclass:: Function
:members:

Profiler
^^^^^^^^

Autograd includes a profiler that lets you inspect the cost of different
operators inside your model - both on the CPU and GPU. There are two modes
implemented at the moment - CPU-only using :class:`~torch.autograd.profiler.profile`.
and nvprof based (registers both CPU and GPU activity) using
:class:`~torch.autograd.profiler.emit_nvtx`.

.. autoclass:: torch.autograd.profiler.profile
:members:

.. autoclass:: torch.autograd.profiler.emit_nvtx
:members:

.. autofunction:: torch.autograd.profiler.load_nvprof
59 changes: 59 additions & 0 deletions docs/0.4.0/_sources/bottleneck.rst.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
torch.utils.bottleneck
======================

.. currentmodule:: torch.utils.bottleneck

`torch.utils.bottleneck` is a tool that can be used as an initial step for
debugging bottlenecks in your program. It summarizes runs of your script with
the Python profiler and PyTorch's autograd profiler.

Run it on the command line with

::

python -m torch.utils.bottleneck /path/to/source/script.py [args]

where [args] are any number of arguments to `script.py`, or run
``python -m torch.utils.bottleneck -h`` for more usage instructions.

.. warning::
Because your script will be profiled, please ensure that it exits in a
finite amount of time.

.. warning::
Due to the asynchronous nature of CUDA kernels, when running against
CUDA code, the cProfile output and CPU-mode autograd profilers may
not show correct timings: the reported CPU time reports the amount of time
used to launch the kernels but does not include the time the kernel
spent executing on a GPU unless the operation does a synchronize.
Ops that do synchronize appear to be extremely expensive under regular
CPU-mode profilers.
In these case where timings are incorrect, the CUDA-mode autograd profiler
may be helpful.

.. note::
To decide which (CPU-only-mode or CUDA-mode) autograd profiler output to
look at, you should first check if your script is CPU-bound
("CPU total time is much greater than CUDA total time").
If it is CPU-bound, looking at the results of the CPU-mode autograd
profiler will help. If on the other hand your script spends most of its
time executing on the GPU, then it makes sense to start
looking for responsible CUDA operators in the output of the CUDA-mode
autograd profiler.

Of course the reality is much more complicated and your script might not be
in one of those two extremes depending on the part of the model you're
evaluating. If the profiler outputs don't help, you could try looking at
the result of :func:`torch.autograd.profiler.emit_nvtx()` with ``nvprof``.
However, please take into account that the NVTX overhead is very high and
often gives a heavily skewed timeline.

.. warning::
If you are profiling CUDA code, the first profiler that ``bottleneck`` runs
(cProfile) will include the CUDA startup time (CUDA buffer allocation cost)
in its time reporting. This should not matter if your bottlenecks result
in code much slower than the CUDA startup time.

For more complicated uses of the profilers (like in a multi-GPU case),
please see https://docs.python.org/3/library/profile.html
or :func:`torch.autograd.profiler.profile()` for more information.
6 changes: 6 additions & 0 deletions docs/0.4.0/_sources/checkpoint.rst.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
torch.utils.checkpoint
======================

.. currentmodule:: torch.utils.checkpoint
.. autofunction:: checkpoint
.. autofunction:: checkpoint_sequential
11 changes: 11 additions & 0 deletions docs/0.4.0/_sources/cpp_extension.rst.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
torch.utils.cpp_extension
=========================

.. currentmodule:: torch.utils.cpp_extension
.. autofunction:: CppExtension
.. autofunction:: CUDAExtension
.. autofunction:: BuildExtension
.. autofunction:: load
.. autofunction:: include_paths
.. autofunction:: check_compiler_abi_compatibility
.. autofunction:: verify_ninja_availability
55 changes: 55 additions & 0 deletions docs/0.4.0/_sources/cuda.rst.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
torch.cuda
===================================

.. currentmodule:: torch.cuda

.. automodule:: torch.cuda
:members:

Random Number Generator
-------------------------
.. autofunction:: get_rng_state
.. autofunction:: set_rng_state
.. autofunction:: manual_seed
.. autofunction:: manual_seed_all
.. autofunction:: seed
.. autofunction:: seed_all
.. autofunction:: initial_seed


Communication collectives
-------------------------

.. autofunction:: torch.cuda.comm.broadcast

.. autofunction:: torch.cuda.comm.broadcast_coalesced

.. autofunction:: torch.cuda.comm.reduce_add

.. autofunction:: torch.cuda.comm.scatter

.. autofunction:: torch.cuda.comm.gather

Streams and events
------------------

.. autoclass:: Stream
:members:

.. autoclass:: Event
:members:

Memory management
-----------------
.. autofunction:: empty_cache
.. autofunction:: memory_allocated
.. autofunction:: max_memory_allocated
.. autofunction:: memory_cached
.. autofunction:: max_memory_cached

NVIDIA Tools Extension (NVTX)
-----------------------------

.. autofunction:: torch.cuda.nvtx.mark
.. autofunction:: torch.cuda.nvtx.range_push
.. autofunction:: torch.cuda.nvtx.range_pop
14 changes: 14 additions & 0 deletions docs/0.4.0/_sources/data.rst.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
torch.utils.data
===================================

.. automodule:: torch.utils.data
.. autoclass:: Dataset
.. autoclass:: TensorDataset
.. autoclass:: ConcatDataset
.. autoclass:: DataLoader
.. autoclass:: torch.utils.data.sampler.Sampler
.. autoclass:: torch.utils.data.sampler.SequentialSampler
.. autoclass:: torch.utils.data.sampler.RandomSampler
.. autoclass:: torch.utils.data.sampler.SubsetRandomSampler
.. autoclass:: torch.utils.data.sampler.WeightedRandomSampler
.. autoclass:: torch.utils.data.distributed.DistributedSampler
Loading