o ?ei%@sBdZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddZedejdejdfddZedejdejdfddZedejejdfddZedejdejdfddZedejdejdfdd Zd!d"ZdS)#z*Ops for computing common window functions.N) constant_op)dtypes)ops) tensor_util) array_ops)cond)math_ops)nn_ops)special_math_ops)dispatch) tf_exportcCs2|js td|tj|tjd}|jd|S)aRCheck window_length and dtype params. Args: window_length: A scalar value or `Tensor`. dtype: The data type to produce. Must be a floating point type. Returns: window_length converted to a tensor of type int32. Raises: ValueError: If `dtype` is not a floating point type or window_length is not a scalar. -dtype must be a floating point type. Found %sdtyper) is_floating ValueErrorrconvert_to_tensorrint32shapeassert_has_rank) window_lengthrrh/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/ops/signal/window_ops.py _check_params s   rzsignal.kaiser_windowg(@c Cst|d|t||}t|}|dkr$tjdg|dWdStj|t j ddd}tj | |dt j d}tj||d}tj||d}tjd|d}tj||d}|t t |t||}t||t|t|} Wd| S1swY| S)aGenerate a [Kaiser window][kaiser]. Args: window_length: A scalar `Tensor` indicating the window length to generate. beta: Beta parameter for Kaiser window, see reference below. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. [kaiser]: https://docs.scipy.org/doc/numpy/reference/generated/numpy.kaiser.html kaiser_windowrNg?@g?)r name_scoperrconstant_valueronesrcastrfloat32rangesqrtr ZreluZsquareexpr Z bessel_i0e) rbetarnamewindow_length_constZ halflen_floatargonenumwindowrrrr5s2     rz#signal.kaiser_bessel_derived_windowc Cst|d;t||}|d}t|d||d}t|}t|dd|d}tj||dddfdd}Wd|S1sDwY|S) aGenerate a [Kaiser Bessel derived window][kbd]. Args: window_length: A scalar `Tensor` indicating the window length to generate. beta: Beta parameter for Kaiser window. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. [kbd]: https://en.wikipedia.org/wiki/Kaiser_window#Kaiser%E2%80%93Bessel-derived_(KBD)_window kaiser_bessel_derived_windowrrNr)Zaxis) rrrrrZcumsumr#rconcat) rr%rr&ZhalflenZkaiserwZ kaiserw_csumZhalfwr+rrrr,\s   r,zsignal.vorbis_windowc Cst|d5t||}tjt||d}ttjdt ttjtj||d|dd}Wd|S1s>wY|S)aGenerate a [Vorbis power complementary window][vorbis]. Args: window_length: A scalar `Tensor` indicating the window length to generate. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. [vorbis]: https://en.wikipedia.org/wiki/Modified_discrete_cosine_transform#Window_functions vorbis_windowrr?N) rrrrr r"sinnppipow)rrr&r(r+rrrr0xs  r0zsignal.hann_windowTcCst|d|||ddS)aGenerate a [Hann window][hann]. Args: window_length: A scalar `Tensor` indicating the window length to generate. periodic: A bool `Tensor` indicating whether to generate a periodic or symmetric window. Periodic windows are typically used for spectral analysis while symmetric windows are typically used for digital filter design. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. Raises: ValueError: If `dtype` is not a floating point type. [hann]: https://en.wikipedia.org/wiki/Window_function#Hann_and_Hamming_windows hann_windowr1_raised_cosine_windowrperiodicrr&rrrr6s r6zsignal.hamming_windowcCst|d|||ddS)aGenerate a [Hamming][hamming] window. Args: window_length: A scalar `Tensor` indicating the window length to generate. periodic: A bool `Tensor` indicating whether to generate a periodic or symmetric window. Periodic windows are typically used for spectral analysis while symmetric windows are typically used for digital filter design. dtype: The data type to produce. Must be a floating point type. name: An optional name for the operation. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. Raises: ValueError: If `dtype` is not a floating point type. [hamming]: https://en.wikipedia.org/wiki/Window_function#Hann_and_Hamming_windows hamming_windowgHzG?gq= ףp?r7r9rrrr;s r;c svjs tdt|||gtjtjddjdt }|dkr:t j dgdWdSt tj|tjddtj}|jddt d }t j||dd} t t } tjd tjd| | |durt jt dWdStt dfd d fd d WdS1swYdS) aHelper function for computing a raised cosine window. Args: name: Name to use for the scope. default_name: Default name to use for the scope. window_length: A scalar `Tensor` or integer indicating the window length. periodic: A bool `Tensor` indicating whether to generate a periodic or symmetric window. dtype: A floating point `DType`. a: The alpha parameter to the raised cosine window. b: The beta parameter to the raised cosine window. Returns: A `Tensor` of shape `[window_length]` of type `dtype`. Raises: ValueError: If `dtype` is not a floating point type or `window_length` is not scalar or `periodic` is not scalar. r r)rr&rrrNr:r-cstjgdSNr)rrr)rrrrsz'_raised_cosine_window..cstjtdSr<)rr cosr)abcos_argrrrr=s)rrrrrrrrrrrrrrr boolmodr"rZconstantr3r4r>requal) r& default_namerr:rr?r@r'Zevenncountr)r?r@rArrrr8s:       $r8)__doc__numpyr3Ztensorflow.python.frameworkrrrrZtensorflow.python.opsrrrr r Ztensorflow.python.utilr Z tensorflow.python.util.tf_exportr rZadd_dispatch_supportr!rr,r0r6r;r8rrrrsD           %