I also tried np.random.rand(1, 50, 50, 1) (see the question). No luck :(. Do you have any other ideas? I really need this to start working. Thank you!
I also tried np.random.rand(1, 1, 50, 50, 1)50) (see the question). No luck :(. Do you have any other ideas? I really need this to start working. works! Thank you!
image = np.random.rand(1, 1, 50, 50)50)
 works! Thank you!