Program Listing for File instancenorm.h

↰ Return to documentation for file (torch/csrc/api/include/torch/nn/functional/instancenorm.h)

#pragma once

#include <torch/nn/options/instancenorm.h>

namespace torch::nn::functional {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
namespace detail {
inline Tensor instance_norm(
    const Tensor& input,
    const Tensor& running_mean,
    const Tensor& running_var,
    const Tensor& weight,
    const Tensor& bias,
    bool use_input_stats,
    double momentum,
    double eps) {
  return torch::instance_norm(
      input,
      weight,
      bias,
      running_mean,
      running_var,
      use_input_stats,
      momentum,
      eps,
      at::globalContext().userEnabledCuDNN());
}
} // namespace detail
#endif /* DOXYGEN_SHOULD_SKIP_THIS */

inline Tensor instance_norm(
    const Tensor& input,
    const InstanceNormFuncOptions& options = {}) {
  return detail::instance_norm(
      input,
      options.running_mean(),
      options.running_var(),
      options.weight(),
      options.bias(),
      options.use_input_stats(),
      options.momentum(),
      options.eps());
}

} // namespace torch::nn::functional