We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e63cd52 commit 056dfb0Copy full SHA for 056dfb0
lib/nmatrix/shortcuts.rb
@@ -494,11 +494,11 @@ def block_diagonal(*params)
494
block_sizes << b.shape[0]
495
end
496
497
- block_diag_mat = NMatrix.zeros(block_sizes.sum, options)
+ block_diag_mat = NMatrix.zeros(block_sizes.inject(0,:+), options)
498
(0...params.length).each do |n|
499
# First determine the size and position of the n'th block in the block-diagonal matrix
500
block_size = block_sizes[n]
501
- block_pos = block_sizes[0...n].sum
+ block_pos = block_sizes[0...n].inject(0,:+)
502
# populate the n'th block in the block-diagonal matrix
503
(0...block_size).each do |i|
504
(0...block_size).each do |j|
0 commit comments