If You are referring to conv_single_step
, keep in mind that Z
should be a scalar with shape ()
and not (1, 1, 1)
. Please use print(Z.shape)
to fix your code.
Also, print(type(Z))
should not produce numpy.ndarray
in the output. It should be something like numpy.float64
.