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)