I\'d like to convert following pytorch code into keras version. But the shape of x is (?, 256, 256, 1) in keras. When I create canny by np.zeros, the error will occ
np.zeros