dtype 到 np array.dtype

Question

我的意图是在tensorflow中对一个张量使用pywavelet，用句子

pywt.dwt(input,'db1')

错误是

Cannot convert DType to numpy.dtype

输入是来自 MNIST 数据集的 Tensor("Reshape_18:0", shape=(?, 28, 28, 1), dtype=float32)。

我尝试使用句子 pywt.dwt(input.dtype.as_numpy_dtype,'db1') 和错误“invalid array_struct”。

Answer 1

pywt.dwt() 采用一维数组。

但看起来你有一个 4D 数组，所以你想使用 pywt.dwtn() Docs here。