logmeanexp.c File Reference

#include "internal.h"

Include dependency graph for logmeanexp.c:

Go to the source code of this file.

Functions
SEXP	logmeanexp (const SEXP X, const SEXP Drop)

Function Documentation

logmeanexp()

SEXP logmeanexp	(	const SEXP	X,
		const SEXP	Drop
	)

Definition at line 7 of file logmeanexp.c.

                                                {
  int j, n = LENGTH(X);
  int k = *INTEGER(Drop)-1;     // zero-based index
  double *x = REAL(X);
  long double m = R_NegInf;
  long double s = 0;
  for (j = 0; j < n; j++) {
    if (j != k)
      m = (x[j] > m) ? (long double) x[j] : m;
  }
  for (j = 0; j < n; j++) {
    if (j != k)
      s += expl((long double) x[j] - m);
  }
  if (k >= 0 && k < n) n--;
  return ScalarReal(m + log(s/n));
}