jax.scipy.stats.bernoulli.logpmfΒΆ

jax.scipy.stats.bernoulli.logpmf(k, p, loc=0)[source]ΒΆ

Log of the probability mass function at k of the given RV.

LAX-backend implementation of logpmf().

Original docstring below.

Parameters
  • k (array_like) – Quantiles.

  • arg1 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information).

  • arg2 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information).

  • arg3 (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information).

  • .. (array_like) – The shape parameter(s) for the distribution (see docstring of the instance object for more information).

  • loc (array_like, optional) – Location parameter. Default is 0.

Returns

logpmf – Log of the probability mass function evaluated at k.

Return type

array_like