15 lines
244 B
Python
15 lines
244 B
Python
import numpy as np
|
|
|
|
a = np.arange(2 * 3 * 4).reshape(2, 3, 4)
|
|
|
|
b = np.arange(12).reshape(2, 6)
|
|
|
|
c = a + b.reshape(2, 3, 4)
|
|
|
|
c_flattened = c.flatten(copy=True)
|
|
|
|
print("a\n", a)
|
|
print("b\n", b)
|
|
print("c\n", c)
|
|
print("c_flattened\n", c_flattened)
|