C1_W1_Assignment UNQ_C3 need help with tensor operations

see if this helps you