nuka137 cbc234bceb C++ API: torch::nn::BatchNorm1d (#28176) ... Summary: Add torch::nn::BatchNorm1d function/module support for the C++ API. torch::nn::BatchNorm{2,3}d will be added after this PR is merged. Related Issue: https://github.com/pytorch/pytorch/issues/25883 Reviewer: yf225 I would like to discuss about below items. * Necessity of `num_batches_tracked` in `BatchNormImplBase` * `num_batches_tracked` is needed to calculate `momentum` when we do not feed `momentum` argument in Python API. But in C++ API, `momentum` argument has a default value. * `num_batches_tracked` is only used for counting up `BatchNorm1d::foward()` call. I think it is no necessary for user anymore. * The design of `BatchNorm{1,2,3}dOptions` * We have already `BatchNormOptions` used for deprecated `BatchNorm` module. However, it is hard to use it for `BatchNorm{1,2,3}dOptions` because of the arguments disagreement of each modules. * In this PR, I introduce `BatchNormOptionsv2` template class for the `BatchNorm{1,2,3}dOptions`. But I'm not sure this design is good or not. Pull Request resolved: https://github.com/pytorch/pytorch/pull/28176 Differential Revision: D18196843 Pulled By: yf225 fbshipit-source-id: 667e2b5de4150d5776c41b9088c9e6c2ead24cd4		2019-10-29 17:29:42 -07:00
..
api	C++ API: torch::nn::BatchNorm1d (#28176)	2019-10-29 17:29:42 -07:00
common	Trim libshm deps, move tempfile.h to c10 (#17019)	2019-02-13 19:38:35 -08:00
dist_autograd	Exclude more files in torch/csrc/distributed when USE_DISTRIBUTED=0 (#28621)	2019-10-28 08:03:30 -07:00
jit	Switching tests to ProfilingExecutor (rebased)	2019-10-29 11:41:42 -07:00
__init__.py	Add train() / eval() / is_training() to C++ ScriptModule API (#16044)	2019-02-01 13:07:38 -08:00