矩阵相乘 matrix multiply, please??

lonelyair

1. #!/usr/bin/perl
   
2. 
   
3. use strict;
   
4. 
   
5. my $matrix_name;
   
6. my $str =<< 'EOF';
   
7. matrixa
   
8. 12 4
   
9. 10 30
   
10. 
    
11. matrixb
    
12. 5 6
    
13. 1 10
    
14. EOF
    
15. 
    
16. \matrix_read_file();
    
17. my @array = \matrix_multiply(\@{$matrix_name->[0]}, \@{$matrix_name->[1]});
    
18. print @array;
    
19. 
    
20. sub matrix_read_file {
    
21.         while($str) {
    
22.                 chomp $str;
    
23.                 next if $str =~ /^\s*$/;
    
24.                 if ($str =~ /^([A-Za-z]\w*)/) {
    
25.                         $matrix_name = $1;
    
26.                 } else {
    
27.                         my (@row) = split (/\s+/, $str);
    
28.                         push (@{$matrix_name}, \@row);
    
29.                 }
    
30.         }
    
31.         $matrix_name
    
32. }
    
33. 
    
34. sub matrix_multiply {
    
35.         my ($r_mat1, $r_mat2) = @_;
    
36.         my $r_product;
    
37.         my ($r1, $c1) = matrix_count_rows_cols($r_mat1);
    
38.         my ($r2, $c2) = matrix_count_rows_cols($r_mat2);
    
39.         die "Matrixs cannot multiply\n" unless ($c1 == $c2);
    
40.         for (my $I=0; $I<$r1; $I++) {
    
41.                 for (my $J; $J<$c2; $J++) {
    
42.                         my $sum = 0;
    
43.                         for (my $k=0; $k<$c1; $k++) {
    
44.                                 $sum += $r_mat1->[$I][$k] * $r_mat2->[$k][$J];
    
45.                         }
    
46.                         $r_product->[$I][$J] = $sum;
    
47.                 }
    
48.         }
    
49.         $r_product;
    
50. }
    
51. 
    
52. sub matrix_count_rows_cols {
    
53.         my $r_mat = @_;
    
54.         my $num_rows = @$r_mat;
    
55.         my $num_cols = @{$r_mat->[0]};
    
56.         ($num_rows, $num_cols);
    
57. }

复制代码

帮我改改,学的不好.

alexru

1. 
   
2. use warnings;
   
3. use strict;
   
4. 
   
5. use PDL;
   
6. 
   
7. my $a = pdl[[12,4], [10,30]];
   
8. my $b = pdl[[5,6], [1,10]];
   
9. 
   
10. my $c = $a x $b;
    
11. 
    
12. print $c;

复制代码

output

1. 
   
2. [
   
3. [ 64 112]
   
4. [ 80 360]
   
5. ]
   

复制代码

[ 本帖最后由 alexru 于 2007-6-16 17:16 编辑 ]