Rev | Line | |
---|
[37] | 1 | function s = logsumexp(a, dim) |
---|
| 2 | % Returns log(sum(exp(a),dim)) while avoiding numerical underflow. |
---|
| 3 | % Default is dim = 1 (columns). |
---|
| 4 | % logsumexp(a, 2) will sum across rows instead of columns. |
---|
| 5 | % Unlike matlab's "sum", it will not switch the summing direction |
---|
| 6 | % if you provide a row vector. |
---|
| 7 | |
---|
| 8 | % Written by Tom Minka |
---|
| 9 | % (c) Microsoft Corporation. All rights reserved. |
---|
| 10 | |
---|
| 11 | if nargin < 2 |
---|
| 12 | dim = 1; |
---|
| 13 | end |
---|
| 14 | |
---|
| 15 | % subtract the largest in each column |
---|
| 16 | [y, i] = max(a,[],dim); |
---|
| 17 | dims = ones(1,ndims(a)); |
---|
| 18 | dims(dim) = size(a,dim); |
---|
| 19 | a = a - repmat(y, dims); |
---|
| 20 | s = y + log(sum(exp(a),dim)); |
---|
| 21 | i = find(~finite(y)); |
---|
| 22 | if ~isempty(i) |
---|
| 23 | s(i) = y(i); |
---|
| 24 | end |
---|
Note: See
TracBrowser
for help on using the repository browser.