mindscience.sciops.dft.DCT

class mindscience.sciops.dft.DCT(shape, compute_dtype=mstype.float32)[源代码]

在最后一个轴上对实数进行 1D 离散余弦变换。结果应与 scipy.fft.dct() 相同。 参考： Type 2 DCT using N FFT (Makhoul)。

参数：

  • shape (tuple) - 要变换的维度形状，其他维度无需包含。必须是长度为 1 的元组。

  • compute_dtype (mindspore.dtype) - 输入张量的类型。默认：mstype.float32。

输入：

  • a (Tensor) - 要变换的实张量，尾随维度与 shape 对齐。

输出：

  • b (Tensor) - 输出的实张量，尾随维度与 shape 对齐。

样例：

>>> from mindspore import ops
>>> from mindflow.cell import DCT
>>> a = ops.rand((2, 32, 512))
>>> dft_cell = DCT(a.shape[-1:])
>>> b = dft_cell(a)
>>> print(b.shape)
(2, 32, 512)