Reshape the training and test examples

#(≈ 2 lines of code)

train_set_x_flatten = …

test_set_x_flatten = …

YOUR CODE STARTS HERE

train_set_x_flatten = train_set_x_orig.reshape(train_set_x_orig.shape[1]*train_set_x_orig.shape[2]*train_set_x_orig.shape[3],train_set_x_orig.shape[0])

test_set_x_flatten = test_set_x_orig.reshape(test_set_x_orig.shape[1]*test_set_x_orig.shape[2]*test_set_x_orig.shape[3],test_set_x_orig.shape[0])

Check that the first 10 pixels of the second image are in the correct place

assert np.alltrue(train_set_x_flatten[0:10, 1] == [196, 192, 190, 193, 186, 182, 188, 179, 174, 213]), “Wrong solution. Use (X.shape[0], -1).T.”

assert np.alltrue(test_set_x_flatten[0:10, 1] == [115, 110, 111, 137, 129, 129, 155, 146, 145, 159]), “Wrong solution. Use (X.shape[0], -1).T.”

print ("train_set_x_flatten shape: " + str(train_set_x_flatten.shape))

print ("train_set_y shape: " + str(train_set_y.shape))

print ("test_set_x_flatten shape: " + str(test_set_x_flatten.shape))

print ("test_set_y shape: " + str(test_set_y.shape))

got this error

AssertionError Traceback (most recent call last)

in

7 test_set_x_flatten = test_set_x_orig.reshape(test_set_x_orig.shape[1]*test_set_x_orig.shape[2]*test_set_x_orig.shape[3],test_set_x_orig.shape[0]).T

8 # Check that the first 10 pixels of the second image are in the correct place

----> 9 assert np.alltrue(train_set_x_flatten[0:10, 1] == [196, 192, 190, 193, 186, 182, 188, 179, 174, 213]), “Wrong solution. Use (X.shape[0], -1).T.”

10 assert np.alltrue(test_set_x_flatten[0:10, 1] == [115, 110, 111, 137, 129, 129, 155, 146, 145, 159]), “Wrong solution. Use (X.shape[0], -1).T.”

11

AssertionError: Wrong solution. Use (X.shape[0], -1).T.

can anyone help