subsref.m

张量分析工具

function a = subsref(t,s)
%SUBSREF Subscripted reference for a ktensor.
%
%   Examples
%   X = ktensor([3; 2], rand(4,2), rand(5,2), rand(3,2));
%   X.lambda returns the lambda array ([3;2]).
%   X.U returns a cell array of 3 matrices.
%   X.U{1} returns the matrix corresponding to the first mode.
%   X(2,3,1) calculates and returns that single element of A.
%
%   See also KTENSOR.
%
%MATLAB Tensor Toolbox.
%Copyright 2007, Sandia Corporation. 

% This is the MATLAB Tensor Toolbox by Brett Bader and Tamara Kolda. 
% http://csmr.ca.sandia.gov/~tgkolda/TensorToolbox.
% Copyright (2007) Sandia Corporation. Under the terms of Contract
% DE-AC04-94AL85000, there is a non-exclusive license for use of this
% work by or on behalf of the U.S. Government. Export of this data may
% require a license from the United States Government.
% The full license terms can be found in tensor_toolbox/LICENSE.txt
% $Id: subsref.m,v 1.7 2007/01/10 01:27:30 bwbader Exp $

switch s(1).type    
    case '.'
        switch s(1).subs
            case 'lambda'
                a = tt_subsubsref(t.lambda,s);
            case {'u','U'}
                a = tt_subsubsref(t.u,s);
            otherwise
                error(['No such field: ', s(1).subs]);
        end
    case '()'
        a = 0;
        for k = 1 : length(t.lambda)
            b = t.lambda(k);
            for i = 1 : length(s.subs)
                b = b * t.u{i}(s.subs{i},k);
            end
            a  = a + b;
        end
    case '{}'
        a = subsref(t.u,s);
    otherwise
        error('Invalid subsref.');
end