\;j[ddlZddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z d d gZdd Zdd Z dd Z ddZy)N)_C_ops)in_dynamic_mode)check_variable_and_dtype) LayerHelper)fft_c2cfft_c2rfft_r2c) is_complexstftistftc0|dvrtd|dt|tr|dkrtd|dt|tr|dkrtd|dtrI||j|kDrtd|d |j|d t j ||||Sd }t|d gd |t|fit}|jd }|j|}|j|d|i|||dd|i|S)a Slice the N-dimensional (where N >= 1) input into (overlapping) frames. Args: x (Tensor): The input data which is a N-dimensional (where N >= 1) Tensor with shape `[..., seq_length]` or `[seq_length, ...]`. frame_length (int): Length of the frame and `0 < frame_length <= x.shape[axis]`. hop_length (int): Number of steps to advance between adjacent frames and `0 < hop_length`. axis (int, optional): Specify the axis to operate on the input Tensors. Its value should be 0(the first dimension) or -1(the last dimension). If not specified, the last axis is used by default. Returns: The output frames tensor with shape `[..., frame_length, num_frames]` if `axis==-1`, otherwise `[num_frames, frame_length, ...]` where `num_frames = 1 + (x.shape[axis] - frame_length) // hop_length` Examples: .. code-block:: python >>> import paddle >>> from paddle import signal >>> # 1D >>> x = paddle.arange(8) >>> y0 = signal.frame(x, frame_length=4, hop_length=2, axis=-1) >>> print(y0) Tensor(shape=[4, 3], dtype=int64, place=Place(cpu), stop_gradient=True, [[0, 2, 4], [1, 3, 5], [2, 4, 6], [3, 5, 7]]) >>> y1 = signal.frame(x, frame_length=4, hop_length=2, axis=0) >>> print(y1) Tensor(shape=[3, 4], dtype=int64, place=Place(cpu), stop_gradient=True, [[0, 1, 2, 3], [2, 3, 4, 5], [4, 5, 6, 7]]) >>> # 2D >>> x0 = paddle.arange(16).reshape([2, 8]) >>> y0 = signal.frame(x0, frame_length=4, hop_length=2, axis=-1) >>> print(y0) Tensor(shape=[2, 4, 3], dtype=int64, place=Place(cpu), stop_gradient=True, [[[0 , 2 , 4 ], [1 , 3 , 5 ], [2 , 4 , 6 ], [3 , 5 , 7 ]], [[8 , 10, 12], [9 , 11, 13], [10, 12, 14], [11, 13, 15]]]) >>> x1 = paddle.arange(16).reshape([8, 2]) >>> y1 = signal.frame(x1, frame_length=4, hop_length=2, axis=0) >>> print(y1.shape) [3, 4, 2] >>> # > 2D >>> x0 = paddle.arange(32).reshape([2, 2, 8]) >>> y0 = signal.frame(x0, frame_length=4, hop_length=2, axis=-1) >>> print(y0.shape) [2, 2, 4, 3] >>> x1 = paddle.arange(32).reshape([8, 2, 2]) >>> y1 = signal.frame(x1, frame_length=4, hop_length=2, axis=0) >>> print(y1.shape) [3, 4, 2, 2] rUnexpected axis: . It should be 0 or -1.rzUnexpected frame_length: #. It should be an positive integer.Unexpected hop_length: zKAttribute frame_length should be less equal than sequence length, but got (z) > (z).framex)int32int64float16float32float64input_param_namedtypeX) frame_length hop_lengthaxisOuttypeinputsattrsoutputs) ValueError isinstanceintrshaperrrrlocals input_dtype"create_variable_for_type_inference append_op) rr!r"r#nameop_typehelperrouts VG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/signal.pyrrsaT 7,TF2IJKK lC (LA,=' ~5X Y   j# &*/%j\1T U   !''$- '(>qwwt}oRA ||A|Z>> sG W11""C"877e7D8 ,( CL   Jc|dvrtd|dt|tr|dkrtd|dd}trt j |||}|St |dgd |t|fit}|jd }|j| }|j|d |i||d d|i|S)a Reconstructs a tensor consisted of overlap added sequences from input frames. Args: x (Tensor): The input data which is a N-dimensional (where N >= 2) Tensor with shape `[..., frame_length, num_frames]` or `[num_frames, frame_length ...]`. hop_length (int): Number of steps to advance between adjacent frames and `0 < hop_length <= frame_length`. axis (int, optional): Specify the axis to operate on the input Tensors. Its value should be 0(the first dimension) or -1(the last dimension). If not specified, the last axis is used by default. Returns: The output frames tensor with shape `[..., seq_length]` if `axis==-1`, otherwise `[seq_length, ...]` where `seq_length = (n_frames - 1) * hop_length + frame_length` Examples: .. code-block:: python >>> import paddle >>> from paddle.signal import overlap_add >>> # 2D >>> x0 = paddle.arange(16).reshape([8, 2]) >>> print(x0) Tensor(shape=[8, 2], dtype=int64, place=Place(cpu), stop_gradient=True, [[0 , 1 ], [2 , 3 ], [4 , 5 ], [6 , 7 ], [8 , 9 ], [10, 11], [12, 13], [14, 15]]) >>> y0 = overlap_add(x0, hop_length=2, axis=-1) >>> print(y0) Tensor(shape=[10], dtype=int64, place=Place(cpu), stop_gradient=True, [0 , 2 , 5 , 9 , 13, 17, 21, 25, 13, 15]) >>> x1 = paddle.arange(16).reshape([2, 8]) >>> print(x1) Tensor(shape=[2, 8], dtype=int64, place=Place(cpu), stop_gradient=True, [[0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 ], [8 , 9 , 10, 11, 12, 13, 14, 15]]) >>> y1 = overlap_add(x1, hop_length=2, axis=0) >>> print(y1) Tensor(shape=[10], dtype=int64, place=Place(cpu), stop_gradient=True, [0 , 1 , 10, 12, 14, 16, 18, 20, 14, 15]) >>> # > 2D >>> x0 = paddle.arange(32).reshape([2, 1, 8, 2]) >>> y0 = overlap_add(x0, hop_length=2, axis=-1) >>> print(y0.shape) [2, 1, 10] >>> x1 = paddle.arange(32).reshape([2, 8, 1, 2]) >>> y1 = overlap_add(x1, hop_length=2, axis=0) >>> print(y1.shape) [10, 1, 2] rrrrrr overlap_addr)rrrrruint16rrr )r"r#r$r%) r*r+r,rrr9rrr.r/r0r1)rr"r#r2r3r5r4rs r6r9r9sL 7,TF2IJKK j# &*/%j\1T U  G  J5" J !  I   W11""C"877e7D8!+T:CL  Jr7c t|j} | dvs Jd| | dk(r|jd}|t|dz}|dkDs Jd|d||}t r5d|cxkr|jd ksnJd |jd d |dd|cxkr|ksnJd |d |d|>t|jdk(rt||k(s:Jd |d|jdt j |f|j}||kr>||z dz} ||z | z } t jjj|| | gd}|ra|dvs Jd|d|dz} t jjj|jd | | g|djd }t|||d }|jgd}t j||}|rdnd}t|r |rJdt|st!|dd |d|| }nt#|dd |d| }|jgd}| dk(r|j%d|S)!a@ Short-time Fourier transform (STFT). The STFT computes the discrete Fourier transforms (DFT) of short overlapping windows of the input using this formula: .. math:: X_t[f] = \sum_{n = 0}^{N-1} \text{window}[n]\ x[t \times H + n]\ e^{-{2 \pi j f n}/{N}} Where: - :math:`t`: The :math:`t`-th input window. - :math:`f`: Frequency :math:`0 \leq f < \text{n_fft}` for `onesided=False`, or :math:`0 \leq f < \lfloor \text{n_fft} / 2 \rfloor + 1` for `onesided=True`. - :math:`N`: Value of `n_fft`. - :math:`H`: Value of `hop_length`. Args: x (Tensor): The input data which is a 1-dimensional or 2-dimensional Tensor with shape `[..., seq_length]`. It can be a real-valued or a complex Tensor. n_fft (int): The number of input samples to perform Fourier transform. hop_length (int, optional): Number of steps to advance between adjacent windows and `0 < hop_length`. Default: `None` (treated as equal to `n_fft//4`) win_length (int, optional): The size of window. Default: `None` (treated as equal to `n_fft`) window (Tensor, optional): A 1-dimensional tensor of size `win_length`. It will be center padded to length `n_fft` if `win_length < n_fft`. Default: `None` ( treated as a rectangle window with value equal to 1 of size `win_length`). center (bool, optional): Whether to pad `x` to make that the :math:`t \times hop\_length` at the center of :math:`t`-th frame. Default: `True`. pad_mode (str, optional): Choose padding pattern when `center` is `True`. See `paddle.nn.functional.pad` for all padding options. Default: `"reflect"` normalized (bool, optional): Control whether to scale the output by `1/sqrt(n_fft)`. Default: `False` onesided (bool, optional): Control whether to return half of the Fourier transform output that satisfies the conjugate symmetry condition when input is a real-valued tensor. It can not be `True` if input is a complex tensor. Default: `True` name (str, optional): The default value is None. Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name`. Returns: The complex STFT output tensor with shape `[..., n_fft//2 + 1, num_frames]` (real-valued input and `onesided` is `True`) or `[..., n_fft, num_frames]` (`onesided` is `False`) Examples: .. code-block:: python >>> import paddle >>> from paddle.signal import stft >>> # real-valued input >>> x = paddle.randn([8, 48000], dtype=paddle.float64) >>> y1 = stft(x, n_fft=512) >>> print(y1.shape) [8, 257, 376] >>> y2 = stft(x, n_fft=512, onesided=False) >>> print(y2.shape) [8, 512, 376] >>> # complex input >>> x = paddle.randn([8, 48000], dtype=paddle.float64) + \ ... paddle.randn([8, 48000], dtype=paddle.float64)*1j >>> print(x.shape) [8, 48000] >>> print(x.dtype) paddle.complex128 >>> y1 = stft(x, n_fft=512, center=False, onesided=False) >>> print(y1.shape) [8, 512, 372] )rz9x should be a 1D or 2D real tensor, but got rank of x is rrNz"hop_length should be > 0, but got .rz"n_fft should be in (0, seq_length( )], but got "win_length should be in (0, n_fft(z8expected a 1D window tensor of size equal to win_length(z), but got window with shape r-rr<constantpadmode)rBreflectz5pad_mode should be "reflect" or "constant", but got "z".NLC)rDrE data_format)rr!r"r#rr<rpermorthobackwardzBonesided should be False when input or window is a complex Tensor.T)rnr#normforwardonesidedr2rrNr#rOrPr2)lenr- unsqueezer,rpaddleonesrnn functionalrDsqueezer transposemultiplyr r rsqueeze_)rn_fftr" win_lengthwindowcenterpad_mode normalizedrQr2x_rankpad_left pad_right pad_lengthx_framesrOr5s r6r r sn\F  L C6(KL  { KKN!_ >M? |1MM>   $ $ R / }Lq Q R $ J%M +E7,zl!LM   "s6{j'@  Ej\Qnouo{o{n||} ~  @J=@EJ&1,J&1 %%)) 9-J*    PC8*B O P aZ II $ $ KKOZ( % '"+ qu"MH!! "Hx0H 7jD( P O P  a= $RdDt  --Y- 'C { Q Jr7c 0t|dddgdt|j} | dvs Jd| | dk(r|jd}|t |d z}||}d|cxkr|ksnJd |d |d d|cxkr|ksnJd|d |d |jd} |jd} t rF|r(| |dzdzk(s9Jdj |dzdz| | |k(sJdj || |Gt|jdk(rt||k(sJdj ||j|jtjtjfvrtjntj}tj|f|}||kr>||z dz}||z |z }tjjj|||gd}|j!gd}|rdnd}| r|rJdt#|d d|dd }n;t%|rJd|dur|d d d d d |dzdzf}t'|d d|dd }tj(||j!gd}t+||d }t+tj,tj(||jd| dg!j!ddg|d }|!|r?|d d |dz|dz f}||dz|dz }n |r|dz}nd}|d d |||zf}||||z}t r:|j/j1j3d"kr t5d#||z }| dk(r|j7d|S)$a Inverse short-time Fourier transform (ISTFT). Reconstruct time-domain signal from the giving complex input and window tensor when nonzero overlap-add (NOLA) condition is met: .. math:: \sum_{t = -\infty}^{\infty} \text{window}^2[n - t \times H]\ \neq \ 0, \ \text{for } all \ n Where: - :math:`t`: The :math:`t`-th input window. - :math:`N`: Value of `n_fft`. - :math:`H`: Value of `hop_length`. Result of `istft` expected to be the inverse of `paddle.signal.stft`, but it is not guaranteed to reconstruct a exactly realizable time-domain signal from a STFT complex tensor which has been modified (via masking or otherwise). Therefore, `istft` gives the `[Griffin-Lim optimal estimate] `_ (optimal in a least-squares sense) for the corresponding signal. Args: x (Tensor): The input data which is a 2-dimensional or 3-dimensional **complex** Tensor with shape `[..., n_fft, num_frames]`. n_fft (int): The size of Fourier transform. hop_length (int, optional): Number of steps to advance between adjacent windows from time-domain signal and `0 < hop_length < win_length`. Default: `None` ( treated as equal to `n_fft//4`) win_length (int, optional): The size of window. Default: `None` (treated as equal to `n_fft`) window (Tensor, optional): A 1-dimensional tensor of size `win_length`. It will be center padded to length `n_fft` if `win_length < n_fft`. It should be a real-valued tensor if `return_complex` is False. Default: `None`(treated as a rectangle window with value equal to 1 of size `win_length`). center (bool, optional): It means that whether the time-domain signal has been center padded. Default: `True`. normalized (bool, optional): Control whether to scale the output by :math:`1/sqrt(n_{fft})`. Default: `False` onesided (bool, optional): It means that whether the input STFT tensor is a half of the conjugate symmetry STFT tensor transformed from a real-valued signal and `istft` will return a real-valued tensor when it is set to `True`. Default: `True`. length (int, optional): Specify the length of time-domain signal. Default: `None`( treated as the whole length of signal). return_complex (bool, optional): It means that whether the time-domain signal is real-valued. If `return_complex` is set to `True`, `onesided` should be set to `False` cause the output is complex. name (str, optional): The default value is None. Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name`. Returns: A tensor of least squares estimation of the reconstructed signal(s) with shape `[..., seq_length]` Examples: .. code-block:: python >>> import numpy as np >>> import paddle >>> from paddle.signal import stft, istft >>> paddle.seed(0) >>> # STFT >>> x = paddle.randn([8, 48000], dtype=paddle.float64) >>> y = stft(x, n_fft=512) >>> print(y.shape) [8, 257, 376] >>> # ISTFT >>> x_ = istft(y, n_fft=512) >>> print(x_.shape) [8, 48000] >>> np.allclose(x, x_) True r complex64 complex128r )r<zr@rrzQfft_size should be equal to n_fft // 2 + 1({}) when onesided is True, but got {}.zIfft_size should be equal to n_fft({}) when onesided is False, but got {}.zZexpected a 1D window tensor of size equal to win_length({}), but got window with shape {}.rArBrCrIrJrLrMzSonesided should be False when input(output of istft) or window is a complex Tensor.FrRzGData type of window should not be complex when return_complex is False.)rr"r#)r repeat_timesgdy=zAbort istft because Nonzero Overlap Add (NOLA) condition failed. For more information about NOLA constraint please see `scipy.signal.check_NOLA`(https://docs.scipy.org/doc/scipy/reference/generated/scipy.signal.check_NOLA.html).)rrSr-rTr,rformatrrUrrirrVrWrXrDrZrr r r[r9tileabsminitemr*r\)rr]r"r^r_r`rbrQlengthreturn_complexr2rcn_framesfft_size window_dtyperdrerOr5window_envelopstarts r6r r s]rQk<%@'J \F  O FfXNO  { KKN!_   J$*$W 0 L TUVW$ J%M +E7,zl!LMwwr{Hwwr{H EQJN* bii Q * E! Zaax !    "s6{j'@ g n n    @ww6>>6+;+;<< NN  J= EEJ&1,J&1 %%)) 9-J*    A!7jD a ` a Tu4P   U T U  u !Q(%1*q.(()ATu4P //#v & 0 0  1 C  *2 C! ++ooff-77:"A  )Q    N~ a%1*%1* 556C+UaZUaZ=IN QJEE!UUV^++,'?^//1557<<>F s    C { Q Jr7)rN)NNNTrFFTN) NNNTFTNFN)rUrpaddle.frameworkrbase.data_feederrbase.layer_helperrfftrr r tensor.attributer __all__rr9r r r7r6rs,6***(    pfbP     nh     Xr7