Func Class — pytorch Architecture

Source Code

aten/src/ATen/functorch/BatchRulesBinaryOps.cpp lines 15–25

template <typename F, F Func, typename... ExtraArgs>
static Tensor _binary_pointwise_batch_rule(
    const Tensor& tensor, std::optional<int64_t> tensor_batch_dim,
    const Tensor& other, std::optional<int64_t> other_batch_dim,
    ExtraArgs... extra_args) {

  auto [tensor_, other_]= _binary_pointwise_helper(
      tensor, tensor_batch_dim, other, other_batch_dim);

  return Func(tensor_, std::move(other_), std::forward<ExtraArgs>(extra_args)...);
}