Dtype to nparray.dtype

Question

Мое намерение состоит в том, чтобы использовать pywavelet на тензор в тензорном потоке, с предложением

pywt.dwt(input,'db1')

Ошибка была

Cannot convert DType to numpy.dtype

Вход является Tensor("Reshape_18:0", shape=(?, 28, 28, 1), dtype=float32) из набора данных MNIST.

Я пытаюсь с предложением pywt.dwt(input.dtype.as_numpy_dtype,'db1') и ошибка "неверный массив_структуры".

Answer 1

pywt.dwt() принимает 1D-массив.

Но похоже, что у вас есть 4D-массив, поэтому вы захотите использовать pywt.dwtn() Документы здесь.