We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 370d5fd commit 16124fcCopy full SHA for 16124fc
matrix/kronecker_product.py
@@ -33,8 +33,8 @@ def kronecker_product(
33
>>> kronecker_product([[1, 2]], [[5, 6], [7, 8]])
34
[[5, 6, 10, 12], [7, 8, 14, 16]]
35
36
- >>> kronecker_product([[1, 2, 3], [4, 5, 6]], [[5, 6], [7, 8]])
37
- [[5, 6, 10, 12, 15, 18], [7, 8, 14, 16, 21, 24], [20, 24, 25, 30, 30, 36], [28, 32, 35, 40, 42, 48]]
+ >>> kronecker_product([[1, 2], [4, 5]], [[5, 6], [7, 8]])
+ [[5, 6, 10, 12], [7, 8, 14, 16], [20, 24, 25, 30], [28, 32, 35, 40]]
38
39
>>> kronecker_product([1, 2], [[5, 6], [7, 8]])
40
Traceback (most recent call last):
@@ -64,7 +64,9 @@ def kronecker_product(
64
for c_index_b in range(cols_matrix_b):
65
result[r_index_a * rows_matrix_b + r_index_b][
66
c_index_a * cols_matrix_b + c_index_b
67
- ] = matrix_a[r_index_a][c_index_a] * matrix_b[r_index_b][c_index_b]
+ ] = (
68
+ matrix_a[r_index_a][c_index_a] * matrix_b[r_index_b][c_index_b]
69
+ )
70
71
return result
72
0 commit comments