目录
torch /////////////////////////////////////// -tree -L 1
.
├── autograd
├── backends
├── _C.cpython-36m-x86_64-linux-gnu.so
├── contrib
├── cuda
├── distributed
├── distributions
├── _dl.cpython-36m-x86_64-linux-gnu.so
├── for_onnx
├── functional.py
├── hub.py
├── init.py
├── init.pyi
├── jit
├── _jit_internal.py
├── lib
├── multiprocessing
├── nn
├── onnx
├── _ops.py
├── optim
├── pycache
├── random.py
├── serialization.py
├── share
├── _six.py
├── sparse
├── _storage_docs.py
├── storage.py
├── _tensor_docs.py
├── tensor.py
├── _tensor_str.py
├── testing
├── _thnn
├── _torch_docs.py
├── utils
├── _utils_internal.py
├── _utils.py
└── version.py
19 directories, 20 files
torch /////////////////////////////////////// -tree -L 2
.
├── autograd
│ ├── anomaly_mode.py
│ ├── function.py
│ ├── _functions
│ ├── gradcheck.py
│ ├── grad_mode.py
│ ├── init.py
│ ├── profiler.py
│ ├── pycache
│ └── variable.py
├── backends
│ ├── cuda
│ ├── cudnn
│ ├── init.py
│ ├── mkl
│ └── pycache
├── _C.cpython-36m-x86_64-linux-gnu.so
├── contrib
│ ├── _graph_vis.py
│ ├── init.py
│ ├── pycache
│ └── _tensorboard_vis.py
├── cuda
│ ├── comm.py
│ ├── error.py
│ ├── init.py
│ ├── nccl.py
│ ├── nvtx.py
│ ├── profiler.py
│ ├── pycache
│ ├── random.py
│ ├── sparse.py
│ ├── streams.py
│ └── _utils.py
├── distributed
│ ├── deprecated
│ ├── distributed_c10d.py
│ ├── init.py
│ ├── launch.py
│ ├── pycache
│ └── rendezvous.py
├── distributions
│ ├── bernoulli.py
│ ├── beta.py
│ ├── binomial.py
│ ├── categorical.py
│ ├── cauchy.py
│ ├── chi2.py
│ ├── constraint_registry.py
│ ├── constraints.py
│ ├── dirichlet.py
│ ├── distribution.py
│ ├── exp_family.py
│ ├── exponential.py
│ ├── fishersnedecor.py
│ ├── gamma.py
│ ├── geometric.py
│ ├── gumbel.py
│ ├── half_cauchy.py
│ ├── half_normal.py
│ ├── independent.py
│ ├── init.py
│ ├── kl.py
│ ├── laplace.py
│ ├── logistic_normal.py
│ ├── log_normal.py
│ ├── lowrank_multivariate_normal.py
│ ├── multinomial.py
│ ├── multivariate_normal.py
│ ├── negative_binomial.py
│ ├── normal.py
│ ├── one_hot_categorical.py
│ ├── pareto.py
│ ├── poisson.py
│ ├── pycache
│ ├── relaxed_bernoulli.py
│ ├── relaxed_categorical.py
│ ├── studentT.py
│ ├── transformed_distribution.py
│ ├── transforms.py
│ ├── uniform.py
│ ├── utils.py
│ └── weibull.py
├── _dl.cpython-36m-x86_64-linux-gnu.so
├── for_onnx
│ ├── init.py
│ └── pycache
├── functional.py
├── hub.py
├── init.py
├── init.pyi
├── jit
│ ├── annotations.py
│ ├── batchop.py
│ ├── frontend.py
│ ├── init.py
│ ├── pycache
│ └── supported_ops.py
├── _jit_internal.py
├── lib
│ ├── include
│ ├── libc10.so
│ ├── libcaffe2_detectron_ops.so
│ ├── libcaffe2_module_test_dynamic.so
│ ├── libcaffe2_observers.so
│ ├── libcaffe2.so
│ ├── libonnxifi_dummy.so
│ ├── libonnxifi.so
│ ├── libshm.so
│ ├── libtorch_python.so
│ ├── libtorch.so
│ ├── libtorch.so.1
│ ├── THCUNN.h
│ ├── THNN.h
│ └── torch_shm_manager
├── multiprocessing
│ ├── init.py
│ ├── pool.py
│ ├── pycache
│ ├── queue.py
│ ├── reductions.py
│ └── spawn.py
├── nn
│ ├── backends
│ ├── functional.py
│ ├── _functions
│ ├── grad.py
│ ├── init.py
│ ├── init.py
│ ├── modules
│ ├── parallel
│ ├── parameter.py
│ ├── pycache
│ ├── _reduction.py
│ ├── utils
│ └── _VF.py
├── onnx
│ ├── init.py
│ ├── operators.py
│ ├── pycache
│ ├── symbolic.py
│ └── utils.py
├── _ops.py
├── optim
│ ├── adadelta.py
│ ├── adagrad.py
│ ├── adamax.py
│ ├── adam.py
│ ├── asgd.py
│ ├── init.py
│ ├── lbfgs.py
│ ├── lr_scheduler.py
│ ├── optimizer.py
│ ├── pycache
│ ├── rmsprop.py
│ ├── rprop.py
│ ├── sgd.py
│ └── sparse_adam.py
├── pycache
│ ├── functional.cpython-36.pyc
│ ├── hub.cpython-36.pyc
│ ├── init.cpython-36.pyc
│ ├── _jit_internal.cpython-36.pyc
│ ├── _ops.cpython-36.pyc
│ ├── random.cpython-36.pyc
│ ├── serialization.cpython-36.pyc
│ ├── _six.cpython-36.pyc
│ ├── storage.cpython-36.pyc
│ ├── _storage_docs.cpython-36.pyc
│ ├── tensor.cpython-36.pyc
│ ├── _tensor_docs.cpython-36.pyc
│ ├── _tensor_str.cpython-36.pyc
│ ├── _torch_docs.cpython-36.pyc
│ ├── _utils.cpython-36.pyc
│ ├── _utils_internal.cpython-36.pyc
│ └── version.cpython-36.pyc
├── random.py
├── serialization.py
├── share
│ └── cmake
├── _six.py
├── sparse
│ ├── init.py
│ └── pycache
├── _storage_docs.py
├── storage.py
├── _tensor_docs.py
├── tensor.py
├── _tensor_str.py
├── testing
│ ├── init.py
│ └── pycache
├── _thnn
│ ├── init.py
│ ├── pycache
│ └── utils.py
├── _torch_docs.py
├── utils
│ ├── backcompat
│ ├── bottleneck
│ ├── checkpoint.py
│ ├── collect_env.py
│ ├── cpp_extension.py
│ ├── _cpp_extension_versioner.py
│ ├── data
│ ├── dlpack.py
│ ├── ffi
│ ├── file_baton.py
│ ├── hooks.py
│ ├── init.py
│ ├── model_zoo.py
│ └── pycache
├── _utils_internal.py
├── _utils.py
└── version.py
51 directories, 164 files
torch /////////////////////////////////////// -tree -L 3
.
├── autograd
│ ├── anomaly_mode.py
│ ├── function.py
│ ├── _functions
│ │ ├── init.py
│ │ ├── pycache
│ │ ├── tensor.py
│ │ └── utils.py
│ ├── gradcheck.py
│ ├── grad_mode.py
│ ├── init.py
│ ├── profiler.py
│ ├── pycache
│ │ ├── anomaly_mode.cpython-36.pyc
│ │ ├── function.cpython-36.pyc
│ │ ├── gradcheck.cpython-36.pyc
│ │ ├── grad_mode.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── profiler.cpython-36.pyc
│ │ └── variable.cpython-36.pyc
│ └── variable.py
├── backends
│ ├── cuda
│ │ ├── init.py
│ │ └── pycache
│ ├── cudnn
│ │ ├── init.py
│ │ ├── pycache
│ │ └── rnn.py
│ ├── init.py
│ ├── mkl
│ │ ├── init.py
│ │ └── pycache
│ └── pycache
│ └── init.cpython-36.pyc
├── _C.cpython-36m-x86_64-linux-gnu.so
├── contrib
│ ├── _graph_vis.py
│ ├── init.py
│ ├── pycache
│ │ ├── _graph_vis.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ └── _tensorboard_vis.cpython-36.pyc
│ └── _tensorboard_vis.py
├── cuda
│ ├── comm.py
│ ├── error.py
│ ├── init.py
│ ├── nccl.py
│ ├── nvtx.py
│ ├── profiler.py
│ ├── pycache
│ │ ├── comm.cpython-36.pyc
│ │ ├── error.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── nccl.cpython-36.pyc
│ │ ├── nvtx.cpython-36.pyc
│ │ ├── profiler.cpython-36.pyc
│ │ ├── random.cpython-36.pyc
│ │ ├── sparse.cpython-36.pyc
│ │ ├── streams.cpython-36.pyc
│ │ └── _utils.cpython-36.pyc
│ ├── random.py
│ ├── sparse.py
│ ├── streams.py
│ └── _utils.py
├── distributed
│ ├── deprecated
│ │ ├── init.py
│ │ ├── pycache
│ │ └── remote_types.py
│ ├── distributed_c10d.py
│ ├── init.py
│ ├── launch.py
│ ├── pycache
│ │ ├── distributed_c10d.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── launch.cpython-36.pyc
│ │ └── rendezvous.cpython-36.pyc
│ └── rendezvous.py
├── distributions
│ ├── bernoulli.py
│ ├── beta.py
│ ├── binomial.py
│ ├── categorical.py
│ ├── cauchy.py
│ ├── chi2.py
│ ├── constraint_registry.py
│ ├── constraints.py
│ ├── dirichlet.py
│ ├── distribution.py
│ ├── exp_family.py
│ ├── exponential.py
│ ├── fishersnedecor.py
│ ├── gamma.py
│ ├── geometric.py
│ ├── gumbel.py
│ ├── half_cauchy.py
│ ├── half_normal.py
│ ├── independent.py
│ ├── init.py
│ ├── kl.py
│ ├── laplace.py
│ ├── logistic_normal.py
│ ├── log_normal.py
│ ├── lowrank_multivariate_normal.py
│ ├── multinomial.py
│ ├── multivariate_normal.py
│ ├── negative_binomial.py
│ ├── normal.py
│ ├── one_hot_categorical.py
│ ├── pareto.py
│ ├── poisson.py
│ ├── pycache
│ │ ├── bernoulli.cpython-36.pyc
│ │ ├── beta.cpython-36.pyc
│ │ ├── binomial.cpython-36.pyc
│ │ ├── categorical.cpython-36.pyc
│ │ ├── cauchy.cpython-36.pyc
│ │ ├── chi2.cpython-36.pyc
│ │ ├── constraint_registry.cpython-36.pyc
│ │ ├── constraints.cpython-36.pyc
│ │ ├── dirichlet.cpython-36.pyc
│ │ ├── distribution.cpython-36.pyc
│ │ ├── exp_family.cpython-36.pyc
│ │ ├── exponential.cpython-36.pyc
│ │ ├── fishersnedecor.cpython-36.pyc
│ │ ├── gamma.cpython-36.pyc
│ │ ├── geometric.cpython-36.pyc
│ │ ├── gumbel.cpython-36.pyc
│ │ ├── half_cauchy.cpython-36.pyc
│ │ ├── half_normal.cpython-36.pyc
│ │ ├── independent.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── kl.cpython-36.pyc
│ │ ├── laplace.cpython-36.pyc
│ │ ├── logistic_normal.cpython-36.pyc
│ │ ├── log_normal.cpython-36.pyc
│ │ ├── lowrank_multivariate_normal.cpython-36.pyc
│ │ ├── multinomial.cpython-36.pyc
│ │ ├── multivariate_normal.cpython-36.pyc
│ │ ├── negative_binomial.cpython-36.pyc
│ │ ├── normal.cpython-36.pyc
│ │ ├── one_hot_categorical.cpython-36.pyc
│ │ ├── pareto.cpython-36.pyc
│ │ ├── poisson.cpython-36.pyc
│ │ ├── relaxed_bernoulli.cpython-36.pyc
│ │ ├── relaxed_categorical.cpython-36.pyc
│ │ ├── studentT.cpython-36.pyc
│ │ ├── transformed_distribution.cpython-36.pyc
│ │ ├── transforms.cpython-36.pyc
│ │ ├── uniform.cpython-36.pyc
│ │ ├── utils.cpython-36.pyc
│ │ └── weibull.cpython-36.pyc
│ ├── relaxed_bernoulli.py
│ ├── relaxed_categorical.py
│ ├── studentT.py
│ ├── transformed_distribution.py
│ ├── transforms.py
│ ├── uniform.py
│ ├── utils.py
│ └── weibull.py
├── _dl.cpython-36m-x86_64-linux-gnu.so
├── for_onnx
│ ├── init.py
│ └── pycache
│ └── init.cpython-36.pyc
├── functional.py
├── hub.py
├── init.py
├── init.pyi
├── jit
│ ├── annotations.py
│ ├── batchop.py
│ ├── frontend.py
│ ├── init.py
│ ├── pycache
│ │ ├── annotations.cpython-36.pyc
│ │ ├── batchop.cpython-36.pyc
│ │ ├── frontend.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ └── supported_ops.cpython-36.pyc
│ └── supported_ops.py
├── _jit_internal.py
├── lib
│ ├── include
│ │ ├── ATen
│ │ ├── c10
│ │ ├── caffe2
│ │ ├── pybind11
│ │ ├── TH
│ │ └── torch
│ ├── libc10.so
│ ├── libcaffe2_detectron_ops.so
│ ├── libcaffe2_module_test_dynamic.so
│ ├── libcaffe2_observers.so
│ ├── libcaffe2.so
│ ├── libonnxifi_dummy.so
│ ├── libonnxifi.so
│ ├── libshm.so
│ ├── libtorch_python.so
│ ├── libtorch.so
│ ├── libtorch.so.1
│ ├── THCUNN.h
│ ├── THNN.h
│ └── torch_shm_manager
├── multiprocessing
│ ├── init.py
│ ├── pool.py
│ ├── pycache
│ │ ├── init.cpython-36.pyc
│ │ ├── pool.cpython-36.pyc
│ │ ├── queue.cpython-36.pyc
│ │ ├── reductions.cpython-36.pyc
│ │ └── spawn.cpython-36.pyc
│ ├── queue.py
│ ├── reductions.py
│ └── spawn.py
├── nn
│ ├── backends
│ │ ├── backend.py
│ │ ├── init.py
│ │ ├── pycache
│ │ └── thnn.py
│ ├── functional.py
│ ├── _functions
│ │ ├── init.py
│ │ ├── pycache
│ │ ├── thnn
│ │ └── vision.py
│ ├── grad.py
│ ├── init.py
│ ├── init.py
│ ├── modules
│ │ ├── activation.py
│ │ ├── adaptive.py
│ │ ├── batchnorm.py
│ │ ├── container.py
│ │ ├── conv.py
│ │ ├── distance.py
│ │ ├── dropout.py
│ │ ├── fold.py
│ │ ├── init.py
│ │ ├── instancenorm.py
│ │ ├── linear.py
│ │ ├── loss.py
│ │ ├── module.py
│ │ ├── normalization.py
│ │ ├── padding.py
│ │ ├── pixelshuffle.py
│ │ ├── pooling.py
│ │ ├── pycache
│ │ ├── rnn.py
│ │ ├── sparse.py
│ │ ├── upsampling.py
│ │ └── utils.py
│ ├── parallel
│ │ ├── data_parallel.py
│ │ ├── deprecated
│ │ ├── distributed_cpu.py
│ │ ├── distributed.py
│ │ ├── _functions.py
│ │ ├── init.py
│ │ ├── parallel_apply.py
│ │ ├── pycache
│ │ ├── replicate.py
│ │ └── scatter_gather.py
│ ├── parameter.py
│ ├── pycache
│ │ ├── functional.cpython-36.pyc
│ │ ├── grad.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── parameter.cpython-36.pyc
│ │ ├── _reduction.cpython-36.pyc
│ │ └── _VF.cpython-36.pyc
│ ├── _reduction.py
│ ├── utils
│ │ ├── clip_grad.py
│ │ ├── convert_parameters.py
│ │ ├── init.py
│ │ ├── pycache
│ │ ├── rnn.py
│ │ ├── spectral_norm.py
│ │ └── weight_norm.py
│ └── _VF.py
├── onnx
│ ├── init.py
│ ├── operators.py
│ ├── pycache
│ │ ├── init.cpython-36.pyc
│ │ ├── operators.cpython-36.pyc
│ │ ├── symbolic.cpython-36.pyc
│ │ └── utils.cpython-36.pyc
│ ├── symbolic.py
│ └── utils.py
├── _ops.py
├── optim
│ ├── adadelta.py
│ ├── adagrad.py
│ ├── adamax.py
│ ├── adam.py
│ ├── asgd.py
│ ├── init.py
│ ├── lbfgs.py
│ ├── lr_scheduler.py
│ ├── optimizer.py
│ ├── pycache
│ │ ├── adadelta.cpython-36.pyc
│ │ ├── adagrad.cpython-36.pyc
│ │ ├── adamax.cpython-36.pyc
│ │ ├── adam.cpython-36.pyc
│ │ ├── asgd.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── lbfgs.cpython-36.pyc
│ │ ├── lr_scheduler.cpython-36.pyc
│ │ ├── optimizer.cpython-36.pyc
│ │ ├── rmsprop.cpython-36.pyc
│ │ ├── rprop.cpython-36.pyc
│ │ ├── sgd.cpython-36.pyc
│ │ └── sparse_adam.cpython-36.pyc
│ ├── rmsprop.py
│ ├── rprop.py
│ ├── sgd.py
│ └── sparse_adam.py
├── pycache
│ ├── functional.cpython-36.pyc
│ ├── hub.cpython-36.pyc
│ ├── init.cpython-36.pyc
│ ├── _jit_internal.cpython-36.pyc
│ ├── _ops.cpython-36.pyc
│ ├── random.cpython-36.pyc
│ ├── serialization.cpython-36.pyc
│ ├── _six.cpython-36.pyc
│ ├── storage.cpython-36.pyc
│ ├── _storage_docs.cpython-36.pyc
│ ├── tensor.cpython-36.pyc
│ ├── _tensor_docs.cpython-36.pyc
│ ├── _tensor_str.cpython-36.pyc
│ ├── _torch_docs.cpython-36.pyc
│ ├── _utils.cpython-36.pyc
│ ├── _utils_internal.cpython-36.pyc
│ └── version.cpython-36.pyc
├── random.py
├── serialization.py
├── share
│ └── cmake
│ ├── ATen
│ ├── Caffe2
│ ├── Gloo
│ └── Torch
├── _six.py
├── sparse
│ ├── init.py
│ └── pycache
│ └── init.cpython-36.pyc
├── _storage_docs.py
├── storage.py
├── _tensor_docs.py
├── tensor.py
├── _tensor_str.py
├── testing
│ ├── init.py
│ └── pycache
│ └── init.cpython-36.pyc
├── _thnn
│ ├── init.py
│ ├── pycache
│ │ ├── init.cpython-36.pyc
│ │ └── utils.cpython-36.pyc
│ └── utils.py
├── _torch_docs.py
├── utils
│ ├── backcompat
│ │ ├── init.py
│ │ └── pycache
│ ├── bottleneck
│ │ ├── init.py
│ │ ├── main.py
│ │ └── pycache
│ ├── checkpoint.py
│ ├── collect_env.py
│ ├── cpp_extension.py
│ ├── _cpp_extension_versioner.py
│ ├── data
│ │ ├── dataloader.py
│ │ ├── dataset.py
│ │ ├── distributed.py
│ │ ├── init.py
│ │ ├── pycache
│ │ └── sampler.py
│ ├── dlpack.py
│ ├── ffi
│ │ ├── init.py
│ │ └── pycache
│ ├── file_baton.py
│ ├── hooks.py
│ ├── init.py
│ ├── model_zoo.py
│ └── pycache
│ ├── checkpoint.cpython-36.pyc
│ ├── collect_env.cpython-36.pyc
│ ├── cpp_extension.cpython-36.pyc
│ ├── _cpp_extension_versioner.cpython-36.pyc
│ ├── dlpack.cpython-36.pyc
│ ├── file_baton.cpython-36.pyc
│ ├── hooks.cpython-36.pyc
│ ├── init.cpython-36.pyc
│ └── model_zoo.cpython-36.pyc
├── _utils_internal.py
├── _utils.py
└── version.py
77 directories, 335 files
torch //////////////////////////////////////// -tree
.
├── autograd
│ ├── anomaly_mode.py
│ ├── function.py
│ ├── _functions
│ │ ├── init.py
│ │ ├── pycache
│ │ │ ├── init.cpython-36.pyc
│ │ │ ├── tensor.cpython-36.pyc
│ │ │ └── utils.cpython-36.pyc
│ │ ├── tensor.py
│ │ └── utils.py
│ ├── gradcheck.py
│ ├── grad_mode.py
│ ├── init.py
│ ├── profiler.py
│ ├── pycache
│ │ ├── anomaly_mode.cpython-36.pyc
│ │ ├── function.cpython-36.pyc
│ │ ├── gradcheck.cpython-36.pyc
│ │ ├── grad_mode.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── profiler.cpython-36.pyc
│ │ └── variable.cpython-36.pyc
│ └── variable.py
├── backends
│ ├── cuda
│ │ ├── init.py
│ │ └── pycache
│ │ └── init.cpython-36.pyc
│ ├── cudnn
│ │ ├── init.py
│ │ ├── pycache
│ │ │ ├── init.cpython-36.pyc
│ │ │ └── rnn.cpython-36.pyc
│ │ └── rnn.py
│ ├── init.py
│ ├── mkl
│ │ ├── init.py
│ │ └── pycache
│ │ └── init.cpython-36.pyc
│ └── pycache
│ └── init.cpython-36.pyc
├── _C.cpython-36m-x86_64-linux-gnu.so
├── contrib
│ ├── _graph_vis.py
│ ├── init.py
│ ├── pycache
│ │ ├── _graph_vis.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ └── _tensorboard_vis.cpython-36.pyc
│ └── _tensorboard_vis.py
├── cuda
│ ├── comm.py
│ ├── error.py
│ ├── init.py
│ ├── nccl.py
│ ├── nvtx.py
│ ├── profiler.py
│ ├── pycache
│ │ ├── comm.cpython-36.pyc
│ │ ├── error.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── nccl.cpython-36.pyc
│ │ ├── nvtx.cpython-36.pyc
│ │ ├── profiler.cpython-36.pyc
│ │ ├── random.cpython-36.pyc
│ │ ├── sparse.cpython-36.pyc
│ │ ├── streams.cpython-36.pyc
│ │ └── _utils.cpython-36.pyc
│ ├── random.py
│ ├── sparse.py
│ ├── streams.py
│ └── _utils.py
├── distributed
│ ├── deprecated
│ │ ├── init.py
│ │ ├── pycache
│ │ │ ├── init.cpython-36.pyc
│ │ │ └── remote_types.cpython-36.pyc
│ │ └── remote_types.py
│ ├── distributed_c10d.py
│ ├── init.py
│ ├── launch.py
│ ├── pycache
│ │ ├── distributed_c10d.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── launch.cpython-36.pyc
│ │ └── rendezvous.cpython-36.pyc
│ └── rendezvous.py
├── distributions
│ ├── bernoulli.py
│ ├── beta.py
│ ├── binomial.py
│ ├── categorical.py
│ ├── cauchy.py
│ ├── chi2.py
│ ├── constraint_registry.py
│ ├── constraints.py
│ ├── dirichlet.py
│ ├── distribution.py
│ ├── exp_family.py
│ ├── exponential.py
│ ├── fishersnedecor.py
│ ├── gamma.py
│ ├── geometric.py
│ ├── gumbel.py
│ ├── half_cauchy.py
│ ├── half_normal.py
│ ├── independent.py
│ ├── init.py
│ ├── kl.py
│ ├── laplace.py
│ ├── logistic_normal.py
│ ├── log_normal.py
│ ├── lowrank_multivariate_normal.py
│ ├── multinomial.py
│ ├── multivariate_normal.py
│ ├── negative_binomial.py
│ ├── normal.py
│ ├── one_hot_categorical.py
│ ├── pareto.py
│ ├── poisson.py
│ ├── pycache
│ │ ├── bernoulli.cpython-36.pyc
│ │ ├── beta.cpython-36.pyc
│ │ ├── binomial.cpython-36.pyc
│ │ ├── categorical.cpython-36.pyc
│ │ ├── cauchy.cpython-36.pyc
│ │ ├── chi2.cpython-36.pyc
│ │ ├── constraint_registry.cpython-36.pyc
│ │ ├── constraints.cpython-36.pyc
│ │ ├── dirichlet.cpython-36.pyc
│ │ ├── distribution.cpython-36.pyc
│ │ ├── exp_family.cpython-36.pyc
│ │ ├── exponential.cpython-36.pyc
│ │ ├── fishersnedecor.cpython-36.pyc
│ │ ├── gamma.cpython-36.pyc
│ │ ├── geometric.cpython-36.pyc
│ │ ├── gumbel.cpython-36.pyc
│ │ ├── half_cauchy.cpython-36.pyc
│ │ ├── half_normal.cpython-36.pyc
│ │ ├── independent.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── kl.cpython-36.pyc
│ │ ├── laplace.cpython-36.pyc
│ │ ├── logistic_normal.cpython-36.pyc
│ │ ├── log_normal.cpython-36.pyc
│ │ ├── lowrank_multivariate_normal.cpython-36.pyc
│ │ ├── multinomial.cpython-36.pyc
│ │ ├── multivariate_normal.cpython-36.pyc
│ │ ├── negative_binomial.cpython-36.pyc
│ │ ├── normal.cpython-36.pyc
│ │ ├── one_hot_categorical.cpython-36.pyc
│ │ ├── pareto.cpython-36.pyc
│ │ ├── poisson.cpython-36.pyc
│ │ ├── relaxed_bernoulli.cpython-36.pyc
│ │ ├── relaxed_categorical.cpython-36.pyc
│ │ ├── studentT.cpython-36.pyc
│ │ ├── transformed_distribution.cpython-36.pyc
│ │ ├── transforms.cpython-36.pyc
│ │ ├── uniform.cpython-36.pyc
│ │ ├── utils.cpython-36.pyc
│ │ └── weibull.cpython-36.pyc
│ ├── relaxed_bernoulli.py
│ ├── relaxed_categorical.py
│ ├── studentT.py
│ ├── transformed_distribution.py
│ ├── transforms.py
│ ├── uniform.py
│ ├── utils.py
│ └── weibull.py
├── _dl.cpython-36m-x86_64-linux-gnu.so
├── for_onnx
│ ├── init.py
│ └── pycache
│ └── init.cpython-36.pyc
├── functional.py
├── hub.py
├── init.py
├── init.pyi
├── jit
│ ├── annotations.py
│ ├── batchop.py
│ ├── frontend.py
│ ├── init.py
│ ├── pycache
│ │ ├── annotations.cpython-36.pyc
│ │ ├── batchop.cpython-36.pyc
│ │ ├── frontend.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ └── supported_ops.cpython-36.pyc
│ └── supported_ops.py
├── _jit_internal.py
├── lib
│ ├── include
│ │ ├── ATen
│ │ │ ├── AccumulateType.h
│ │ │ ├── Allocator.h
│ │ │ ├── ArrayRef.h
│ │ │ ├── ATen.h
│ │ │ ├── Backend.h
│ │ │ ├── Backtrace.h
│ │ │ ├── CheckGenerator.h
│ │ │ ├── Config.h
│ │ │ ├── Context.h
│ │ │ ├── core
│ │ │ │ ├── alias_info.h
│ │ │ │ ├── ATenGeneral.h
│ │ │ │ ├── aten_interned_strings.h
│ │ │ │ ├── Backtrace.h
│ │ │ │ ├── blob.h
│ │ │ │ ├── context_base.h
│ │ │ │ ├── DefaultDtype.h
│ │ │ │ ├── Deprecated.h
│ │ │ │ ├── DimVector.h
│ │ │ │ ├── Formatting.h
│ │ │ │ ├── functional.h
│ │ │ │ ├── function_schema.h
│ │ │ │ ├── Generator.h
│ │ │ │ ├── Half.h
│ │ │ │ ├── interned_strings_class.h
│ │ │ │ ├── interned_strings.h
│ │ │ │ ├── ivalue.h
│ │ │ │ ├── jit_type.h
│ │ │ │ ├── LegacyDeviceTypeInit.h
│ │ │ │ ├── LegacyTypeDispatch.h
│ │ │ │ ├── Macros.h
│ │ │ │ ├── Range.h
│ │ │ │ ├── Reduction.h
│ │ │ │ ├── Scalar.h
│ │ │ │ ├── ScalarType.h
│ │ │ │ ├── SparseTensorRef.h
│ │ │ │ ├── TensorAccessor.h
│ │ │ │ ├── Tensor.h
│ │ │ │ ├── TensorImpl.h
│ │ │ │ ├── TensorMethods.h
│ │ │ │ ├── TensorOptions.h
│ │ │ │ ├── thread_pool.h
│ │ │ │ ├── Type.h
│ │ │ │ ├── typeid.h
│ │ │ │ ├── UndefinedTensorImpl.h
│ │ │ │ ├── VariableHooksInterface.h
│ │ │ │ └── WrapDimMinimal.h
│ │ │ ├── cpu
│ │ │ │ ├── FlushDenormal.h
│ │ │ │ └── vml.h
│ │ │ ├── CPUApplyUtils.h
│ │ │ ├── CPUByteType.h
│ │ │ ├── CPUCharType.h
│ │ │ ├── CPUDoubleType.h
│ │ │ ├── CPUFixedAllocator.h
│ │ │ ├── CPUFloatType.h
│ │ │ ├── CPUGeneral.h
│ │ │ ├── CPUGenerator.h
│ │ │ ├── CPUHalfType.h
│ │ │ ├── CPUIntType.h
│ │ │ ├── CPULongType.h
│ │ │ ├── CPUShortType.h
│ │ │ ├── CPUTypeDefault.h
│ │ │ ├── cuda
│ │ │ │ ├── Array.h
│ │ │ │ ├── ATenCUDAGeneral.h
│ │ │ │ ├── CUDAApplyUtils.cuh
│ │ │ │ ├── CUDAConfig.h
│ │ │ │ ├── CUDAContext.h
│ │ │ │ ├── CUDADevice.h
│ │ │ │ ├── CUDAEvent.h
│ │ │ │ ├── CUDAGuard.h
│ │ │ │ ├── CUDAMultiStreamGuard.h
│ │ │ │ ├── CUDAStream.h
│ │ │ │ ├── CUDATensorMethods.cuh
│ │ │ │ ├── CUDATypeDefault.h
│ │ │ │ ├── CUDAUtils.h
│ │ │ │ ├── _curand_mtgp32_host.h
│ │ │ │ ├── detail
│ │ │ │ │ ├── CUDAGuardImpl.h
│ │ │ │ │ ├── CUDAHooks.h
│ │ │ │ │ ├── IndexUtils.cuh
│ │ │ │ │ ├── KernelUtils.h
│ │ │ │ │ ├── OffsetCalculator.cuh
│ │ │ │ │ └── TensorInfo.cuh
│ │ │ │ ├── Exceptions.h
│ │ │ │ ├── NumericLimits.cuh
│ │ │ │ └── PinnedMemoryAllocator.h
│ │ │ ├── CUDAByteType.h
│ │ │ ├── CUDACharType.h
│ │ │ ├── CUDADoubleType.h
│ │ │ ├── CUDAFloatType.h
│ │ │ ├── CUDAGenerator.h
│ │ │ ├── CUDAHalfType.h
│ │ │ ├── CUDAIntType.h
│ │ │ ├── CUDALongType.h
│ │ │ ├── CUDAShortType.h
│ │ │ ├── cudnn
│ │ │ │ ├── cudnn-wrapper.h
│ │ │ │ ├── Descriptors.h
│ │ │ │ ├── Exceptions.h
│ │ │ │ ├── Handle.h
│ │ │ │ ├── Handles.h
│ │ │ │ ├── Types.h
│ │ │ │ └── Utils.h
│ │ │ ├── detail
│ │ │ │ ├── ComplexHooksInterface.h
│ │ │ │ ├── CPUGuardImpl.h
│ │ │ │ ├── CUDAHooksInterface.h
│ │ │ │ ├── FunctionTraits.h
│ │ │ │ ├── HIPHooksInterface.h
│ │ │ │ └── ScalarTypeConversions.h
│ │ │ ├── DeviceGuard.h
│ │ │ ├── Device.h
│ │ │ ├── DimVector.h
│ │ │ ├── Dispatch.h
│ │ │ ├── div_rtn.h
│ │ │ ├── DLConvertor.h
│ │ │ ├── dlpack.h
│ │ │ ├── ExpandUtils.h
│ │ │ ├── Formatting.h
│ │ │ ├── Functions.h
│ │ │ ├── Generator.h
│ │ │ ├── Half.h
│ │ │ ├── InferSize.h
│ │ │ ├── InitialTensorOptions.h
│ │ │ ├── Layout.h
│ │ │ ├── LegacyTHDispatcher.h
│ │ │ ├── LegacyTHDispatch.h
│ │ │ ├── MatrixRef.h
│ │ │ ├── NativeFunctions.h
│ │ │ ├── Parallel.h
│ │ │ ├── RegisterCPU.h
│ │ │ ├── RegisterCUDA.h
│ │ │ ├── Scalar.h
│ │ │ ├── ScalarOps.h
│ │ │ ├── ScalarType.h
│ │ │ ├── SmallVector.h
│ │ │ ├── SparseCPUByteType.h
│ │ │ ├── SparseCPUCharType.h
│ │ │ ├── SparseCPUDoubleType.h
│ │ │ ├── SparseCPUFloatType.h
│ │ │ ├── SparseCPUIntType.h
│ │ │ ├── SparseCPULongType.h
│ │ │ ├── SparseCPUShortType.h
│ │ │ ├── SparseCUDAByteType.h
│ │ │ ├── SparseCUDACharType.h
│ │ │ ├── SparseCUDADoubleType.h
│ │ │ ├── SparseCUDAFloatType.h
│ │ │ ├── SparseCUDAIntType.h
│ │ │ ├── SparseCUDALongType.h
│ │ │ ├── SparseCUDAShortType.h
│ │ │ ├── SparseTensorImpl.h
│ │ │ ├── SparseTensorUtils.h
│ │ │ ├── Storage.h
│ │ │ ├── TensorAccessor.h
│ │ │ ├── TensorGeometry.h
│ │ │ ├── Tensor.h
│ │ │ ├── TensorOperators.h
│ │ │ ├── TensorOptions.h
│ │ │ ├── TensorUtils.h
│ │ │ ├── TypeDefault.h
│ │ │ ├── TypeExtendedInterface.h
│ │ │ ├── Type.h
│ │ │ ├── UndefinedType.h
│ │ │ ├── Utils.h
│ │ │ ├── WrapDimUtils.h
│ │ │ └── WrapDimUtilsMulti.h
│ │ ├── c10
│ │ │ ├── core
│ │ │ │ ├── Allocator.h
│ │ │ │ ├── Backend.h
│ │ │ │ ├── bitcasts.h
│ │ │ │ ├── Layout.h
│ │ │ │ ├── Scalar.h
│ │ │ │ ├── ScalarType.h
│ │ │ │ ├── ScalarTypeUtils.h
│ │ │ │ ├── Storage.h
│ │ │ │ ├── StorageImpl.h
│ │ │ │ ├── TensorTypeId.h
│ │ │ │ └── TensorTypeIdRegistration.h
│ │ │ ├── cuda
│ │ │ │ ├── CUDAException.h
│ │ │ │ ├── CUDAFunctions.h
│ │ │ │ ├── CUDAMacros.h
│ │ │ │ ├── CUDAMathCompat.h
│ │ │ │ └── impl
│ │ │ │ └── CUDATest.h
│ │ │ ├── DeviceGuard.h
│ │ │ ├── Device.h
│ │ │ ├── DeviceType.h
│ │ │ ├── Half.h
│ │ │ ├── Half-inl.h
│ │ │ ├── impl
│ │ │ │ ├── DeviceGuardImplInterface.h
│ │ │ │ ├── FakeGuardImpl.h
│ │ │ │ ├── InlineDeviceGuard.h
│ │ │ │ ├── InlineStreamGuard.h
│ │ │ │ └── VirtualGuardImpl.h
│ │ │ ├── macros
│ │ │ │ ├── cmake_macros.h
│ │ │ │ ├── Export.h
│ │ │ │ └── Macros.h
│ │ │ ├── StreamGuard.h
│ │ │ ├── Stream.h
│ │ │ └── util
│ │ │ ├── AlignOf.h
│ │ │ ├── Array.h
│ │ │ ├── ArrayRef.h
│ │ │ ├── Backtrace.h
│ │ │ ├── C++17.h
│ │ │ ├── Exception.h
│ │ │ ├── Flags.h
│ │ │ ├── flat_hash_map.h
│ │ │ ├── IdWrapper.h
│ │ │ ├── intrusive_ptr.h
│ │ │ ├── LeftRight.h
│ │ │ ├── Logging.h
│ │ │ ├── logging_is_google_glog.h
│ │ │ ├── logging_is_not_google_glog.h
│ │ │ ├── Metaprogramming.h
│ │ │ ├── Optional.h
│ │ │ ├── Registry.h
│ │ │ ├── SmallVector.h
│ │ │ ├── StringUtil.h
│ │ │ ├── string_utils.h
│ │ │ ├── Type.h
│ │ │ ├── typeid.h
│ │ │ ├── TypeList.h
│ │ │ ├── TypeTraits.h
│ │ │ └── UniqueVoidPtr.h
│ │ ├── caffe2
│ │ │ ├── core
│ │ │ │ ├── allocator.h
│ │ │ │ ├── asan.h
│ │ │ │ ├── blob.h
│ │ │ │ ├── blob_serialization.h
│ │ │ │ ├── blob_serializer_base.h
│ │ │ │ ├── blob_stats.h
│ │ │ │ ├── common_cudnn.h
│ │ │ │ ├── common_gpu.h
│ │ │ │ ├── common.h
│ │ │ │ ├── common_omp.h
│ │ │ │ ├── context_base.h
│ │ │ │ ├── context_gpu.h
│ │ │ │ ├── context.h
│ │ │ │ ├── cudnn_wrappers.h
│ │ │ │ ├── db.h
│ │ │ │ ├── event_cpu.h
│ │ │ │ ├── event.h
│ │ │ │ ├── flags.h
│ │ │ │ ├── graph.h
│ │ │ │ ├── init.h
│ │ │ │ ├── logging.h
│ │ │ │ ├── macros.h
│ │ │ │ ├── memonger.h
│ │ │ │ ├── module.h
│ │ │ │ ├── net_async_base.h
│ │ │ │ ├── net_async_scheduling.h
│ │ │ │ ├── net_async_tracing.h
│ │ │ │ ├── net_dag_utils.h
│ │ │ │ ├── net.h
│ │ │ │ ├── net_simple.h
│ │ │ │ ├── net_simple_refcount.h
│ │ │ │ ├── numa.h
│ │ │ │ ├── observer.h
│ │ │ │ ├── operator_c10wrapper.h
│ │ │ │ ├── operator_gradient.h
│ │ │ │ ├── operator.h
│ │ │ │ ├── operator_schema.h
│ │ │ │ ├── plan_executor.h
│ │ │ │ ├── prof_dag_counters.h
│ │ │ │ ├── qtensor.h
│ │ │ │ ├── qtensor_serialization.h
│ │ │ │ ├── scope_guard.h
│ │ │ │ ├── static_tracepoint_elfx86.h
│ │ │ │ ├── static_tracepoint.h
│ │ │ │ ├── stats.h
│ │ │ │ ├── storage.h
│ │ │ │ ├── tensor.h
│ │ │ │ ├── tensor_impl.h
│ │ │ │ ├── tensor_int8.h
│ │ │ │ ├── THCCachingAllocator_gpu.h
│ │ │ │ ├── timer.h
│ │ │ │ ├── transform.h
│ │ │ │ ├── types.h
│ │ │ │ └── workspace.h
│ │ │ ├── proto
│ │ │ │ ├── caffe2_legacy.pb.h
│ │ │ │ ├── caffe2_pb.h
│ │ │ │ ├── caffe2.pb.h
│ │ │ │ ├── hsm.pb.h
│ │ │ │ ├── metanet.pb.h
│ │ │ │ ├── predictor_consts.pb.h
│ │ │ │ ├── prof_dag.pb.h
│ │ │ │ ├── torch_pb.h
│ │ │ │ └── torch.pb.h
│ │ │ └── utils
│ │ │ ├── bench_utils.h
│ │ │ ├── cast.h
│ │ │ ├── cblas.h
│ │ │ ├── conversions.h
│ │ │ ├── cpuid.h
│ │ │ ├── cpu_neon.h
│ │ │ ├── eigen_utils.h
│ │ │ ├── filler.h
│ │ │ ├── fixed_divisor.h
│ │ │ ├── map_utils.h
│ │ │ ├── math-detail.h
│ │ │ ├── math.h
│ │ │ ├── math_utils.h
│ │ │ ├── murmur_hash3.h
│ │ │ ├── proto_convert.h
│ │ │ ├── proto_utils.h
│ │ │ ├── proto_wrap.h
│ │ │ ├── signal_handler.h
│ │ │ ├── simple_queue.h
│ │ │ ├── smart_tensor_printer.h
│ │ │ ├── string_utils.h
│ │ │ ├── thread_name.h
│ │ │ ├── thread_pool.h
│ │ │ └── zmq_helper.h
│ │ ├── pybind11
│ │ │ ├── attr.h
│ │ │ ├── buffer_info.h
│ │ │ ├── cast.h
│ │ │ ├── chrono.h
│ │ │ ├── common.h
│ │ │ ├── complex.h
│ │ │ ├── detail
│ │ │ │ ├── class.h
│ │ │ │ ├── common.h
│ │ │ │ ├── descr.h
│ │ │ │ ├── init.h
│ │ │ │ ├── internals.h
│ │ │ │ └── typeid.h
│ │ │ ├── eigen.h
│ │ │ ├── embed.h
│ │ │ ├── eval.h
│ │ │ ├── functional.h
│ │ │ ├── iostream.h
│ │ │ ├── numpy.h
│ │ │ ├── operators.h
│ │ │ ├── options.h
│ │ │ ├── pybind11.h
│ │ │ ├── pytypes.h
│ │ │ ├── stl_bind.h
│ │ │ └── stl.h
│ │ ├── TH
│ │ │ ├── generic
│ │ │ │ ├── THBlas.h
│ │ │ │ ├── THLapack.h
│ │ │ │ ├── THStorageCopy.h
│ │ │ │ ├── THStorage.h
│ │ │ │ ├── THTensorConv.h
│ │ │ │ ├── THTensorFastGetSet.hpp
│ │ │ │ ├── THTensor.h
│ │ │ │ ├── THTensor.hpp
│ │ │ │ ├── THTensorLapack.h
│ │ │ │ ├── THTensorMath.h
│ │ │ │ ├── THTensorRandom.h
│ │ │ │ └── THVector.h
│ │ │ ├── THAllocator.h
│ │ │ ├── THBlas.h
│ │ │ ├── THDiskFile.h
│ │ │ ├── THFile.h
│ │ │ ├── THFilePrivate.h
│ │ │ ├── THGeneral.h
│ │ │ ├── THGenerateAllTypes.h
│ │ │ ├── THGenerateByteType.h
│ │ │ ├── THGenerateCharType.h
│ │ │ ├── THGenerateDoubleType.h
│ │ │ ├── THGenerateFloatType.h
│ │ │ ├── THGenerateFloatTypes.h
│ │ │ ├── THGenerateHalfType.h
│ │ │ ├── THGenerateIntType.h
│ │ │ ├── THGenerateIntTypes.h
│ │ │ ├── THGenerateLongType.h
│ │ │ ├── THGenerateShortType.h
│ │ │ ├── THGenerator.hpp
│ │ │ ├── TH.h
│ │ │ ├── THHalf.h
│ │ │ ├── THLapack.h
│ │ │ ├── THLogAdd.h
│ │ │ ├── THMath.h
│ │ │ ├── THMemoryFile.h
│ │ │ ├── THRandom.h
│ │ │ ├── THSize.h
│ │ │ ├── THStorageFunctions.h
│ │ │ ├── THStorageFunctions.hpp
│ │ │ ├── THStorage.h
│ │ │ ├── THTensorApply.h
│ │ │ ├── THTensorDimApply.h
│ │ │ ├── THTensor.h
│ │ │ ├── THTensor.hpp
│ │ │ └── THVector.h
│ │ └── torch
│ │ ├── csrc
│ │ │ ├── api
│ │ │ │ └── include
│ │ │ │ └── torch
│ │ │ │ ├── all.h
│ │ │ │ ├── arg.h
│ │ │ │ ├── cuda.h
│ │ │ │ ├── data
│ │ │ │ │ ├── dataloader.h
│ │ │ │ │ ├── dataloader_options.h
│ │ │ │ │ ├── datasets
│ │ │ │ │ │ ├── base.h
│ │ │ │ │ │ ├── map.h
│ │ │ │ │ │ ├── mnist.h
│ │ │ │ │ │ ├── shared.h
│ │ │ │ │ │ └── tensor.h
│ │ │ │ │ ├── datasets.h
│ │ │ │ │ ├── detail
│ │ │ │ │ │ ├── data_shuttle.h
│ │ │ │ │ │ ├── queue.h
│ │ │ │ │ │ └── sequencers.h
│ │ │ │ │ ├── example.h
│ │ │ │ │ ├── iterator.h
│ │ │ │ │ ├── samplers
│ │ │ │ │ │ ├── base.h
│ │ │ │ │ │ ├── custom_batch_request.h
│ │ │ │ │ │ ├── random.h
│ │ │ │ │ │ ├── sequential.h
│ │ │ │ │ │ ├── serialize.h
│ │ │ │ │ │ └── stream.h
│ │ │ │ │ ├── samplers.h
│ │ │ │ │ ├── transforms
│ │ │ │ │ │ ├── base.h
│ │ │ │ │ │ ├── collate.h
│ │ │ │ │ │ ├── lambda.h
│ │ │ │ │ │ ├── stack.h
│ │ │ │ │ │ └── tensor.h
│ │ │ │ │ ├── transforms.h
│ │ │ │ │ └── worker_exception.h
│ │ │ │ ├── data.h
│ │ │ │ ├── detail
│ │ │ │ │ └── static.h
│ │ │ │ ├── expanding_array.h
│ │ │ │ ├── jit.h
│ │ │ │ ├── nn
│ │ │ │ │ ├── cloneable.h
│ │ │ │ │ ├── init.h
│ │ │ │ │ ├── module.h
│ │ │ │ │ ├── modules
│ │ │ │ │ │ ├── any.h
│ │ │ │ │ │ ├── batchnorm.h
│ │ │ │ │ │ ├── conv.h
│ │ │ │ │ │ ├── dropout.h
│ │ │ │ │ │ ├── embedding.h
│ │ │ │ │ │ ├── functional.h
│ │ │ │ │ │ ├── linear.h
│ │ │ │ │ │ ├── rnn.h
│ │ │ │ │ │ └── sequential.h
│ │ │ │ │ ├── modules.h
│ │ │ │ │ ├── parallel
│ │ │ │ │ │ └── data_parallel.h
│ │ │ │ │ ├── pimpl.h
│ │ │ │ │ └── pimpl-inl.h
│ │ │ │ ├── nn.h
│ │ │ │ ├── optim
│ │ │ │ │ ├── adagrad.h
│ │ │ │ │ ├── adam.h
│ │ │ │ │ ├── lbfgs.h
│ │ │ │ │ ├── optimizer.h
│ │ │ │ │ ├── rmsprop.h
│ │ │ │ │ ├── serialize.h
│ │ │ │ │ └── sgd.h
│ │ │ │ ├── optim.h
│ │ │ │ ├── ordered_dict.h
│ │ │ │ ├── python.h
│ │ │ │ ├── serialize
│ │ │ │ │ ├── archive.h
│ │ │ │ │ ├── input-archive.h
│ │ │ │ │ ├── output-archive.h
│ │ │ │ │ └── tensor.h
│ │ │ │ ├── serialize.h
│ │ │ │ ├── torch.h
│ │ │ │ ├── types.h
│ │ │ │ └── utils.h
│ │ │ ├── autograd
│ │ │ │ ├── anomaly_mode.h
│ │ │ │ ├── autograd.h
│ │ │ │ ├── edge.h
│ │ │ │ ├── engine.h
│ │ │ │ ├── function.h
│ │ │ │ ├── function_hook.h
│ │ │ │ ├── generated
│ │ │ │ │ ├── Functions.h
│ │ │ │ │ ├── python_functions.h
│ │ │ │ │ ├── python_nn_functions_dispatch.h
│ │ │ │ │ ├── python_nn_functions.h
│ │ │ │ │ ├── python_torch_functions_dispatch.h
│ │ │ │ │ ├── python_variable_methods_dispatch.h
│ │ │ │ │ ├── variable_factories.h
│ │ │ │ │ └── VariableType.h
│ │ │ │ ├── grad_mode.h
│ │ │ │ ├── input_buffer.h
│ │ │ │ ├── input_metadata.h
│ │ │ │ ├── profiler.h
│ │ │ │ ├── python_anomaly_mode.h
│ │ │ │ ├── python_cpp_function.h
│ │ │ │ ├── python_engine.h
│ │ │ │ ├── python_function.h
│ │ │ │ ├── python_hook.h
│ │ │ │ ├── python_legacy_variable.h
│ │ │ │ ├── python_variable.h
│ │ │ │ ├── python_variable_indexing.h
│ │ │ │ ├── saved_variable.h
│ │ │ │ ├── symbolic.h
│ │ │ │ ├── type_and_shape.h
│ │ │ │ ├── variable.h
│ │ │ │ ├── VariableTypeUtils.h
│ │ │ │ └── variable_version.h
│ │ │ ├── byte_order.h
│ │ │ ├── copy_utils.h
│ │ │ ├── cuda
│ │ │ │ ├── comm.h
│ │ │ │ ├── cuda_check.h
│ │ │ │ ├── device_set.h
│ │ │ │ ├── Module.h
│ │ │ │ ├── nccl.h
│ │ │ │ ├── override_macros.h
│ │ │ │ ├── python_comm.h
│ │ │ │ ├── python_nccl.h
│ │ │ │ ├── restore_macros.h
│ │ │ │ ├── serialization.h
│ │ │ │ ├── Storage.h
│ │ │ │ ├── Stream.h
│ │ │ │ ├── THCP.h
│ │ │ │ ├── undef_macros.h
│ │ │ │ └── utils.h
│ │ │ ├── DataLoader.h
│ │ │ ├── Device.h
│ │ │ ├── Dtype.h
│ │ │ ├── DynamicTypes.h
│ │ │ ├── Exceptions.h
│ │ │ ├── Generator.h
│ │ │ ├── jit
│ │ │ │ ├── alias_info.h
│ │ │ │ ├── argument_spec.h
│ │ │ │ ├── assertions.h
│ │ │ │ ├── attributes.h
│ │ │ │ ├── autodiff.h
│ │ │ │ ├── code_template.h
│ │ │ │ ├── constants.h
│ │ │ │ ├── custom_operator.h
│ │ │ │ ├── dynamic_dag.h
│ │ │ │ ├── export.h
│ │ │ │ ├── function_schema.h
│ │ │ │ ├── generic_if.h
│ │ │ │ ├── graph_executor.h
│ │ │ │ ├── graph_node_list.h
│ │ │ │ ├── hooks_for_testing.h
│ │ │ │ ├── import.h
│ │ │ │ ├── import_method.h
│ │ │ │ ├── init.h
│ │ │ │ ├── interned_strings_class.h
│ │ │ │ ├── interned_strings.h
│ │ │ │ ├── interpreter.h
│ │ │ │ ├── ir.h
│ │ │ │ ├── ir_views.h
│ │ │ │ ├── ivalue.h
│ │ │ │ ├── named_value.h
│ │ │ │ ├── node_hashing.h
│ │ │ │ ├── operator.h
│ │ │ │ ├── passes
│ │ │ │ │ ├── alias_analysis.h
│ │ │ │ │ ├── batch_mm.h
│ │ │ │ │ ├── canonicalize.h
│ │ │ │ │ ├── canonicalize_ops.h
│ │ │ │ │ ├── common_subexpression_elimination.h
│ │ │ │ │ ├── constant_pooling.h
│ │ │ │ │ ├── constant_propagation.h
│ │ │ │ │ ├── create_autodiff_subgraphs.h
│ │ │ │ │ ├── dead_code_elimination.h
│ │ │ │ │ ├── erase_number_types.h
│ │ │ │ │ ├── graph_fuser.h
│ │ │ │ │ ├── inline_autodiff_subgraphs.h
│ │ │ │ │ ├── inplace_check.h
│ │ │ │ │ ├── loop_unrolling.h
│ │ │ │ │ ├── lower_grad_of.h
│ │ │ │ │ ├── lower_tuples.h
│ │ │ │ │ ├── onnx.h
│ │ │ │ │ ├── peephole.h
│ │ │ │ │ ├── python_print.h
│ │ │ │ │ ├── remove_expands.h
│ │ │ │ │ ├── remove_inplace_ops.h
│ │ │ │ │ ├── requires_grad_analysis.h
│ │ │ │ │ ├── shape_analysis.h
│ │ │ │ │ ├── specialize_undef.h
│ │ │ │ │ └── to_batch.h
│ │ │ │ ├── pybind.h
│ │ │ │ ├── pybind_utils.h
│ │ │ │ ├── python_arg_flatten.h
│ │ │ │ ├── python_ir.h
│ │ │ │ ├── python_tracer.h
│ │ │ │ ├── resource_guard.h
│ │ │ │ ├── scope.h
│ │ │ │ ├── script
│ │ │ │ │ ├── builtin_functions.h
│ │ │ │ │ ├── compiler.h
│ │ │ │ │ ├── error_report.h
│ │ │ │ │ ├── init.h
│ │ │ │ │ ├── jit_exception.h
│ │ │ │ │ ├── lexer.h
│ │ │ │ │ ├── module.h
│ │ │ │ │ ├── parser.h
│ │ │ │ │ ├── parse_string_literal.h
│ │ │ │ │ ├── python_tree_views.h
│ │ │ │ │ ├── tree.h
│ │ │ │ │ └── tree_views.h
│ │ │ │ ├── source_location.h
│ │ │ │ ├── source_range.h
│ │ │ │ ├── stack.h
│ │ │ │ ├── symbolic_variable.h
│ │ │ │ ├── tracer.h
│ │ │ │ ├── tracing_state.h
│ │ │ │ ├── type.h
│ │ │ │ └── variable_tensor_list.h
│ │ │ ├── Layout.h
│ │ │ ├── Module.h
│ │ │ ├── PtrWrapper.h
│ │ │ ├── python_headers.h
│ │ │ ├── PythonTypes.h
│ │ │ ├── serialization.h
│ │ │ ├── Size.h
│ │ │ ├── StorageDefs.h
│ │ │ ├── Storage.h
│ │ │ ├── THP_API.h
│ │ │ ├── THP_export.h
│ │ │ ├── THP.h
│ │ │ ├── TypeInfo.h
│ │ │ ├── Types.h
│ │ │ ├── utils
│ │ │ │ ├── auto_gil.h
│ │ │ │ ├── cuda_enabled.h
│ │ │ │ ├── cuda_lazy_init.h
│ │ │ │ ├── disallow_copy.h
│ │ │ │ ├── functional.h
│ │ │ │ ├── hash.h
│ │ │ │ ├── invalid_arguments.h
│ │ │ │ ├── memory.h
│ │ │ │ ├── numpy_stub.h
│ │ │ │ ├── object_ptr.h
│ │ │ │ ├── pybind.h
│ │ │ │ ├── python_arg_parser.h
│ │ │ │ ├── python_compat.h
│ │ │ │ ├── python_numbers.h
│ │ │ │ ├── python_scalars.h
│ │ │ │ ├── python_strings.h
│ │ │ │ ├── python_stub.h
│ │ │ │ ├── python_tuples.h
│ │ │ │ ├── tempfile.h
│ │ │ │ ├── tensor_apply.h
│ │ │ │ ├── tensor_dtypes.h
│ │ │ │ ├── tensor_flatten.h
│ │ │ │ ├── tensor_layouts.h
│ │ │ │ ├── tensor_list.h
│ │ │ │ ├── tensor_new.h
│ │ │ │ ├── tensor_numpy.h
│ │ │ │ ├── tensor_types.h
│ │ │ │ ├── tuple_parser.h
│ │ │ │ └── variadic.h
│ │ │ ├── utils.h
│ │ │ ├── variable_tensor_functions.h
│ │ │ └── WindowsTorchApiMacro.h
│ │ ├── extension.h
│ │ └── script.h
│ ├── libc10.so
│ ├── libcaffe2_detectron_ops.so
│ ├── libcaffe2_module_test_dynamic.so
│ ├── libcaffe2_observers.so
│ ├── libcaffe2.so
│ ├── libonnxifi_dummy.so
│ ├── libonnxifi.so
│ ├── libshm.so
│ ├── libtorch_python.so
│ ├── libtorch.so
│ ├── libtorch.so.1
│ ├── THCUNN.h
│ ├── THNN.h
│ └── torch_shm_manager
├── multiprocessing
│ ├── init.py
│ ├── pool.py
│ ├── pycache
│ │ ├── init.cpython-36.pyc
│ │ ├── pool.cpython-36.pyc
│ │ ├── queue.cpython-36.pyc
│ │ ├── reductions.cpython-36.pyc
│ │ └── spawn.cpython-36.pyc
│ ├── queue.py
│ ├── reductions.py
│ └── spawn.py
├── nn
│ ├── backends
│ │ ├── backend.py
│ │ ├── init.py
│ │ ├── pycache
│ │ │ ├── backend.cpython-36.pyc
│ │ │ ├── init.cpython-36.pyc
│ │ │ └── thnn.cpython-36.pyc
│ │ └── thnn.py
│ ├── functional.py
│ ├── _functions
│ │ ├── init.py
│ │ ├── pycache
│ │ │ ├── init.cpython-36.pyc
│ │ │ └── vision.cpython-36.pyc
│ │ ├── thnn
│ │ │ ├── auto_double_backwards.py
│ │ │ ├── auto.py
│ │ │ ├── auto_symbolic.py
│ │ │ ├── fold.py
│ │ │ ├── init.py
│ │ │ ├── normalization.py
│ │ │ ├── pycache
│ │ │ │ ├── auto.cpython-36.pyc
│ │ │ │ ├── auto_double_backwards.cpython-36.pyc
│ │ │ │ ├── auto_symbolic.cpython-36.pyc
│ │ │ │ ├── fold.cpython-36.pyc
│ │ │ │ ├── init.cpython-36.pyc
│ │ │ │ ├── normalization.cpython-36.pyc
│ │ │ │ └── sparse.cpython-36.pyc
│ │ │ └── sparse.py
│ │ └── vision.py
│ ├── grad.py
│ ├── init.py
│ ├── init.py
│ ├── modules
│ │ ├── activation.py
│ │ ├── adaptive.py
│ │ ├── batchnorm.py
│ │ ├── container.py
│ │ ├── conv.py
│ │ ├── distance.py
│ │ ├── dropout.py
│ │ ├── fold.py
│ │ ├── init.py
│ │ ├── instancenorm.py
│ │ ├── linear.py
│ │ ├── loss.py
│ │ ├── module.py
│ │ ├── normalization.py
│ │ ├── padding.py
│ │ ├── pixelshuffle.py
│ │ ├── pooling.py
│ │ ├── pycache
│ │ │ ├── activation.cpython-36.pyc
│ │ │ ├── adaptive.cpython-36.pyc
│ │ │ ├── batchnorm.cpython-36.pyc
│ │ │ ├── container.cpython-36.pyc
│ │ │ ├── conv.cpython-36.pyc
│ │ │ ├── distance.cpython-36.pyc
│ │ │ ├── dropout.cpython-36.pyc
│ │ │ ├── fold.cpython-36.pyc
│ │ │ ├── init.cpython-36.pyc
│ │ │ ├── instancenorm.cpython-36.pyc
│ │ │ ├── linear.cpython-36.pyc
│ │ │ ├── loss.cpython-36.pyc
│ │ │ ├── module.cpython-36.pyc
│ │ │ ├── normalization.cpython-36.pyc
│ │ │ ├── padding.cpython-36.pyc
│ │ │ ├── pixelshuffle.cpython-36.pyc
│ │ │ ├── pooling.cpython-36.pyc
│ │ │ ├── rnn.cpython-36.pyc
│ │ │ ├── sparse.cpython-36.pyc
│ │ │ ├── upsampling.cpython-36.pyc
│ │ │ └── utils.cpython-36.pyc
│ │ ├── rnn.py
│ │ ├── sparse.py
│ │ ├── upsampling.py
│ │ └── utils.py
│ ├── parallel
│ │ ├── data_parallel.py
│ │ ├── deprecated
│ │ │ ├── distributed_cpu.py
│ │ │ ├── distributed.py
│ │ │ ├── init.py
│ │ │ └── pycache
│ │ │ ├── distributed_cpu.cpython-36.pyc
│ │ │ ├── distributed.cpython-36.pyc
│ │ │ └── init.cpython-36.pyc
│ │ ├── distributed_cpu.py
│ │ ├── distributed.py
│ │ ├── _functions.py
│ │ ├── init.py
│ │ ├── parallel_apply.py
│ │ ├── pycache
│ │ │ ├── data_parallel.cpython-36.pyc
│ │ │ ├── distributed_cpu.cpython-36.pyc
│ │ │ ├── distributed.cpython-36.pyc
│ │ │ ├── _functions.cpython-36.pyc
│ │ │ ├── init.cpython-36.pyc
│ │ │ ├── parallel_apply.cpython-36.pyc
│ │ │ ├── replicate.cpython-36.pyc
│ │ │ └── scatter_gather.cpython-36.pyc
│ │ ├── replicate.py
│ │ └── scatter_gather.py
│ ├── parameter.py
│ ├── pycache
│ │ ├── functional.cpython-36.pyc
│ │ ├── grad.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── parameter.cpython-36.pyc
│ │ ├── _reduction.cpython-36.pyc
│ │ └── _VF.cpython-36.pyc
│ ├── _reduction.py
│ ├── utils
│ │ ├── clip_grad.py
│ │ ├── convert_parameters.py
│ │ ├── init.py
│ │ ├── pycache
│ │ │ ├── clip_grad.cpython-36.pyc
│ │ │ ├── convert_parameters.cpython-36.pyc
│ │ │ ├── init.cpython-36.pyc
│ │ │ ├── rnn.cpython-36.pyc
│ │ │ ├── spectral_norm.cpython-36.pyc
│ │ │ └── weight_norm.cpython-36.pyc
│ │ ├── rnn.py
│ │ ├── spectral_norm.py
│ │ └── weight_norm.py
│ └── _VF.py
├── onnx
│ ├── init.py
│ ├── operators.py
│ ├── pycache
│ │ ├── init.cpython-36.pyc
│ │ ├── operators.cpython-36.pyc
│ │ ├── symbolic.cpython-36.pyc
│ │ └── utils.cpython-36.pyc
│ ├── symbolic.py
│ └── utils.py
├── _ops.py
├── optim
│ ├── adadelta.py
│ ├── adagrad.py
│ ├── adamax.py
│ ├── adam.py
│ ├── asgd.py
│ ├── init.py
│ ├── lbfgs.py
│ ├── lr_scheduler.py
│ ├── optimizer.py
│ ├── pycache
│ │ ├── adadelta.cpython-36.pyc
│ │ ├── adagrad.cpython-36.pyc
│ │ ├── adamax.cpython-36.pyc
│ │ ├── adam.cpython-36.pyc
│ │ ├── asgd.cpython-36.pyc
│ │ ├── init.cpython-36.pyc
│ │ ├── lbfgs.cpython-36.pyc
│ │ ├── lr_scheduler.cpython-36.pyc
│ │ ├── optimizer.cpython-36.pyc
│ │ ├── rmsprop.cpython-36.pyc
│ │ ├── rprop.cpython-36.pyc
│ │ ├── sgd.cpython-36.pyc
│ │ └── sparse_adam.cpython-36.pyc
│ ├── rmsprop.py
│ ├── rprop.py
│ ├── sgd.py
│ └── sparse_adam.py
├── pycache
│ ├── functional.cpython-36.pyc
│ ├── hub.cpython-36.pyc
│ ├── init.cpython-36.pyc
│ ├── _jit_internal.cpython-36.pyc
│ ├── _ops.cpython-36.pyc
│ ├── random.cpython-36.pyc
│ ├── serialization.cpython-36.pyc
│ ├── _six.cpython-36.pyc
│ ├── storage.cpython-36.pyc
│ ├── _storage_docs.cpython-36.pyc
│ ├── tensor.cpython-36.pyc
│ ├── _tensor_docs.cpython-36.pyc
│ ├── _tensor_str.cpython-36.pyc
│ ├── _torch_docs.cpython-36.pyc
│ ├── _utils.cpython-36.pyc
│ ├── _utils_internal.cpython-36.pyc
│ └── version.cpython-36.pyc
├── random.py
├── serialization.py
├── share
│ └── cmake
│ ├── ATen
│ │ └── ATenConfig.cmake
│ ├── Caffe2
│ │ ├── Caffe2Config.cmake
│ │ ├── Caffe2ConfigVersion.cmake
│ │ ├── Caffe2Targets.cmake
│ │ ├── Caffe2Targets-release.cmake
│ │ ├── Modules_CUDA_fix
│ │ │ ├── FindCUDA.cmake
│ │ │ └── upstream
│ │ │ ├── CMakeInitializeConfigs.cmake
│ │ │ ├── FindCUDA
│ │ │ │ ├── make2cmake.cmake
│ │ │ │ ├── parse_cubin.cmake
│ │ │ │ ├── run_nvcc.cmake
│ │ │ │ └── select_compute_arch.cmake
│ │ │ ├── FindCUDA.cmake
│ │ │ ├── FindPackageHandleStandardArgs.cmake
│ │ │ └── FindPackageMessage.cmake
│ │ └── public
│ │ ├── cuda.cmake
│ │ ├── gflags.cmake
│ │ ├── glog.cmake
│ │ ├── mkl.cmake
│ │ ├── mkldnn.cmake
│ │ ├── protobuf.cmake
│ │ ├── threads.cmake
│ │ └── utils.cmake
│ ├── Gloo
│ │ ├── GlooConfig.cmake
│ │ ├── GlooConfigVersion.cmake
│ │ ├── GlooTargets.cmake
│ │ └── GlooTargets-release.cmake
│ └── Torch
│ ├── TorchConfig.cmake
│ └── TorchConfigVersion.cmake
├── _six.py
├── sparse
│ ├── init.py
│ └── pycache
│ └── init.cpython-36.pyc
├── _storage_docs.py
├── storage.py
├── _tensor_docs.py
├── tensor.py
├── _tensor_str.py
├── testing
│ ├── init.py
│ └── pycache
│ └── init.cpython-36.pyc
├── _thnn
│ ├── init.py
│ ├── pycache
│ │ ├── init.cpython-36.pyc
│ │ └── utils.cpython-36.pyc
│ └── utils.py
├── _torch_docs.py
├── utils
│ ├── backcompat
│ │ ├── init.py
│ │ └── pycache
│ │ └── init.cpython-36.pyc
│ ├── bottleneck
│ │ ├── init.py
│ │ ├── main.py
│ │ └── pycache
│ │ ├── init.cpython-36.pyc
│ │ └── main.cpython-36.pyc
│ ├── checkpoint.py
│ ├── collect_env.py
│ ├── cpp_extension.py
│ ├── _cpp_extension_versioner.py
│ ├── data
│ │ ├── dataloader.py
│ │ ├── dataset.py
│ │ ├── distributed.py
│ │ ├── init.py
│ │ ├── pycache
│ │ │ ├── dataloader.cpython-36.pyc
│ │ │ ├── dataset.cpython-36.pyc
│ │ │ ├── distributed.cpython-36.pyc
│ │ │ ├── init.cpython-36.pyc
│ │ │ └── sampler.cpython-36.pyc
│ │ └── sampler.py
│ ├── dlpack.py
│ ├── ffi
│ │ ├── init.py
│ │ └── pycache
│ │ └── init.cpython-36.pyc
│ ├── file_baton.py
│ ├── hooks.py
│ ├── init.py
│ ├── model_zoo.py
│ └── pycache
│ ├── checkpoint.cpython-36.pyc
│ ├── collect_env.cpython-36.pyc
│ ├── cpp_extension.cpython-36.pyc
│ ├── _cpp_extension_versioner.cpython-36.pyc
│ ├── dlpack.cpython-36.pyc
│ ├── file_baton.cpython-36.pyc
│ ├── hooks.cpython-36.pyc
│ ├── init.cpython-36.pyc
│ └── model_zoo.cpython-36.pyc
├── _utils_internal.py
├── _utils.py
└── version.py
122 directories, 1067 files