Mercurial > hg > octave-nkf
annotate scripts/linear-algebra/trace.m @ 11542:695141f1c05c ss-3-3-55
snapshot 3.3.55
author | John W. Eaton <jwe@octave.org> |
---|---|
date | Sat, 15 Jan 2011 04:53:04 -0500 |
parents | fd0a3ac60b0e |
children | 72c96de7a403 |
rev | line source |
---|---|
11523 | 1 ## Copyright (C) 1993-2011 John W. Eaton |
2313 | 2 ## |
3 ## This file is part of Octave. | |
4 ## | |
5 ## Octave is free software; you can redistribute it and/or modify it | |
6 ## under the terms of the GNU General Public License as published by | |
7016 | 7 ## the Free Software Foundation; either version 3 of the License, or (at |
8 ## your option) any later version. | |
2313 | 9 ## |
10 ## Octave is distributed in the hope that it will be useful, but | |
11 ## WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
13 ## General Public License for more details. | |
14 ## | |
15 ## You should have received a copy of the GNU General Public License | |
7016 | 16 ## along with Octave; see the file COPYING. If not, see |
17 ## <http://www.gnu.org/licenses/>. | |
245 | 18 |
3372 | 19 ## -*- texinfo -*- |
11469
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
20 ## @deftypefn {Function File} {} trace (@var{A}) |
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
21 ## Compute the trace of @var{A}, @code{sum (diag (@var{A}))}. |
3372 | 22 ## @end deftypefn |
4 | 23 |
2314 | 24 ## Author: jwe |
25 | |
11469
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
26 function y = trace (A) |
4 | 27 |
28 if (nargin != 1) | |
6046 | 29 print_usage (); |
4 | 30 endif |
31 | |
11469
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
32 if (ndims (A) > 2) |
7415 | 33 error ("trace: only valid on 2-D objects"); |
11469
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
34 elseif (isempty (A)) |
7415 | 35 y = 0; |
11469
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
36 elseif (any (size (A) == 1)) |
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
37 y = A(1); |
1065 | 38 else |
11469
c776f063fefe
Overhaul m-script files to use common variable name between code and documentation.
Rik <octave@nomad.inbox5.com>
parents:
8920
diff
changeset
|
39 y = sum (diag (A)); |
1065 | 40 endif |
4 | 41 |
42 endfunction | |
7411 | 43 |
44 %!assert(trace ([1, 2; 3, 4]) == 5); | |
45 %!assert(trace ([1, 2; 3, 4; 5, 6]) == 5); | |
46 %!assert(trace ([1, 3, 5; 2, 4, 6]) == 5); | |
7418 | 47 %!assert(trace ([]), 0); |
48 %!assert(trace (randn(1,0)), 0); | |
49 %! | |
7411 | 50 %!error trace (); |
51 %!error trace (1, 2); | |
7418 | 52 %!error <only valid on 2-D objects> trace(reshape(1:9,[1,3,3])); |