<SYSTEM_TASK:>
Given the following text description, write Python code to implement the functionality described below step by step
<END_TASK>
<USER_TASK:>
Description:
 saving
checking out how to save current weight matrices and load them back in.
Step1:  first just checking that the flattening and reshaping works as expected
<ASSISTANT_TASK:>
Python Code:
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
Explanation: saving
checking out how to save current weight matrices and load them back in.
End of explanation
test = np.random.randn(11,11,4,100)
test.shape
test_flat = test.flatten()
test_flat.shape
np.savetxt('test.txt', test_flat)
test_back = np.loadtxt('test.txt').reshape((11,11,4,100))
test_back.shape
np.mean(test - test_back) 
Explanation: first just checking that the flattening and reshaping works as expected
End of explanation
<END_TASK>