View source on GitHub
|
Computes log(mean(exp(input_tensor))).
tfp.math.reduce_logmeanexp(
input_tensor, axis=None, keepdims=False, name=None
)
Reduces input_tensor along the dimensions given in axis. Unless
keepdims is true, the rank of the tensor is reduced by 1 for each entry in
axis. If keepdims is true, the reduced dimensions are retained with length
1.
If axis has no entries, all dimensions are reduced, and a tensor with a
single element is returned.
This function is more numerically stable than log(reduce_mean(exp(input))).
It avoids overflows caused by taking the exp of large inputs and underflows
caused by taking the log of small inputs.
Args | |
|---|---|
input_tensor
|
The tensor to reduce. Should have numeric type. |
axis
|
The dimensions to reduce. If None (the default), reduces all
dimensions. Must be in the range [-rank(input_tensor),
rank(input_tensor)).
|
keepdims
|
Boolean. Whether to keep the axis as singleton dimensions.
Default value: False (i.e., squeeze the reduced dimensions).
|
name
|
Python str name prefixed to Ops created by this function.
Default value: None (i.e., 'reduce_logmeanexp').
|
Returns | |
|---|---|
log_mean_exp
|
The reduced tensor. |