cbrt.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_FUN_CBRT_HPP
#define STAN_MATH_REV_FUN_CBRT_HPP
 
#include <stan/math/rev/meta.hpp>
#include <stan/math/rev/core.hpp>
#include <stan/math/prim/fun/cbrt.hpp>
 
namespace stan {
namespace math {
 
inline var cbrt(const var& a) {
  return make_callback_var(cbrt(a.val()), [a](const auto& vi) mutable {
    a.adj() += vi.adj() / (3.0 * vi.val() * vi.val());
  });
}
 
template <typename VarMat, require_var_matrix_t<VarMat>* = nullptr>
inline auto cbrt(const VarMat& a) {
  return make_callback_var(
      a.val().unaryExpr([](const auto x) { return cbrt(x); }),
      [a](const auto& vi) mutable {
        a.adj().array() += vi.adj().array() / (3.0 * vi.val().array().square());
      });
}
 
}  // namespace math
}  // namespace stan
#endif