Skip to content

Commit 056dfb0

Browse files
committed
Fixes #602, missing Array#sum in NMatrix#block_diagonal (MRI only, not JRuby)
1 parent e63cd52 commit 056dfb0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lib/nmatrix/shortcuts.rb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -494,11 +494,11 @@ def block_diagonal(*params)
494494
block_sizes << b.shape[0]
495495
end
496496

497-
block_diag_mat = NMatrix.zeros(block_sizes.sum, options)
497+
block_diag_mat = NMatrix.zeros(block_sizes.inject(0,:+), options)
498498
(0...params.length).each do |n|
499499
# First determine the size and position of the n'th block in the block-diagonal matrix
500500
block_size = block_sizes[n]
501-
block_pos = block_sizes[0...n].sum
501+
block_pos = block_sizes[0...n].inject(0,:+)
502502
# populate the n'th block in the block-diagonal matrix
503503
(0...block_size).each do |i|
504504
(0...block_size).each do |j|

0 commit comments

Comments
 (0)