GNU Octave  3.8.0
A high-level interpreted language, primarily intended for numerical computations, mostly compatible with Matlab
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
smatm3.f
Go to the documentation of this file.
1 c Copyright (C) 2009-2013 VZLU Prague, a.s., Czech Republic
2 c
3 c Author: Jaroslav Hajek <highegg@gmail.com>
4 c
5 c This file is part of Octave.
6 c
7 c Octave is free software; you can redistribute it and/or modify
8 c it under the terms of the GNU General Public License as published by
9 c the Free Software Foundation; either version 3 of the License, or
10 c (at your option) any later version.
11 c
12 c This program is distributed in the hope that it will be useful,
13 c but WITHOUT ANY WARRANTY; without even the implied warranty of
14 c MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 c GNU General Public License for more details.
16 c
17 c You should have received a copy of the GNU General Public License
18 c along with this software; see the file COPYING. If not, see
19 c <http://www.gnu.org/licenses/>.
20 c
21  subroutine smatm3(m,n,k,np,a,b,c)
22 c purpose: a 3-dimensional matrix product.
23 c given a (m,k,np) array a and (k,n,np) array b,
24 c calculates a (m,n,np) array c such that
25 c for i = 1:np
26 c c(:,:,i) = a(:,:,i) * b(:,:,i)
27 c
28 c arguments:
29 c m,n,k (in) the dimensions
30 c np (in) number of multiplications
31 c a (in) a real input array, size (m,k,np)
32 c b (in) a real input array, size (k,n,np)
33 c c (out) a real output array, size (m,n,np)
34  integer m,n,k,np
35  real a(m*k,np),b(k*n,np)
36  real c(m*n,np)
37 
38  real sdot,one,zero
39  parameter(one = 1e0, zero = 0e0)
40  external sdot,sgemv,sgemm
41  integer i
42 
43 c quick return if possible.
44  if (np <= 0) return
45 
46  if (m == 1) then
47  if (n == 1) then
48  do i = 1,np
49  c(1,i) = sdot(k,a(1,i),1,b(1,i),1)
50  end do
51  else
52  do i = 1,np
53  call sgemv("T",k,n,one,b(1,i),k,a(1,i),1,zero,c(1,i),1)
54  end do
55  end if
56  else
57  if (n == 1) then
58  do i = 1,np
59  call sgemv("N",m,k,one,a(1,i),m,b(1,i),1,zero,c(1,i),1)
60  end do
61  else
62  do i = 1,np
63  call sgemm("N","N",m,n,k,
64  + one,a(1,i),m,b(1,i),k,zero,c(1,i),m)
65  end do
66  end if
67  end if
68 
69  end subroutine