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