o ?e'@sdZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZd d Zed e j  dddZdS)zGeneral shape ops for frames.N) constant_op)ops) tensor_util) array_ops)math_ops)util_ops)dispatch) tf_exportc Cst|}t|}t|}|jjdurdS|dur$dg|jjdS|j}d}||}|d|}||ddd} |rl|durl|durU|rUtd| | }n|durl|durl|raJtd||||}|||g| S)z0Infers the shape of the return value of `frame`.Nr)rconstant_valueshapeZndimsas_listmax) signal frame_length frame_steppad_endaxis signal_shape num_framesZ frame_axisouter_dimensionsinner_dimensionsrg/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/ops/signal/shape_ops.py_infer_frame_shapes*        rz signal.frameFc" Cs`t|d||||gtj|dd}tj|dd}tj|dd}tj|dd}|jd|jd|jd|jdt|||||}d d }|t|\} } ||\}} | r{| r{tt | |}t j | ||dgd \} } }| } n"t |}t||}t | |d|d|g\} } }t| g} t| }t|}|rt||j}|jd| | }td|||d| }ttj|d g|jd td|ggtj|d g|jd gd}tj|||d}t|} | |} nttjd|jd d| ||}|t||\}}||}||}| |}t| ||g|gd}t| ||g|gd}tt|t| ||}ttj||jd ||dg}ttj||jd d|g}||}t| } tj|| jd }tj||g| jd } ttj|||dt| | |gd}!|r|!||!WdS1swYdS)a/ Expands `signal`'s `axis` dimension into frames of `frame_length`. Slides a window of size `frame_length` over `signal`'s `axis` dimension with a stride of `frame_step`, replacing the `axis` dimension with `[frames, frame_length]` frames. If `pad_end` is True, window positions that are past the end of the `axis` dimension are padded with `pad_value` until the window moves fully past the end of the dimension. Otherwise, only window positions that fully overlap the `axis` dimension are produced. For example: >>> # A batch size 3 tensor of 9152 audio samples. >>> audio = tf.random.normal([3, 9152]) >>> >>> # Compute overlapping frames of length 512 with a step of 180 (frames overlap >>> # by 332 samples). By default, only 49 frames are generated since a frame >>> # with start position j*180 for j > 48 would overhang the end. >>> frames = tf.signal.frame(audio, 512, 180) >>> frames.shape.assert_is_compatible_with([3, 49, 512]) >>> >>> # When pad_end is enabled, the final two frames are kept (padded with zeros). >>> frames = tf.signal.frame(audio, 512, 180, pad_end=True) >>> frames.shape.assert_is_compatible_with([3, 51, 512]) If the dimension along `axis` is N, and `pad_end=False`, the number of frames can be computed by: ```python num_frames = 1 + (N - frame_size) // frame_step ``` If `pad_end=True`, the number of frames can be computed by: ```python num_frames = -(-N // frame_step) # ceiling division ``` Args: signal: A `[..., samples, ...]` `Tensor`. The rank and dimensions may be unknown. Rank must be at least 1. frame_length: The frame length in samples. An integer or scalar `Tensor`. frame_step: The frame hop size in samples. An integer or scalar `Tensor`. pad_end: Whether to pad the end of `signal` with `pad_value`. pad_value: An optional scalar `Tensor` to use where the input signal does not exist when `pad_end` is True. axis: A scalar integer `Tensor` indicating the axis to frame. Defaults to the last axis. Supports negative values for indexing from the end. name: An optional name for the operation. Returns: A `Tensor` of frames with shape `[..., num_frames, frame_length, ...]`. Raises: ValueError: If `frame_length`, `frame_step`, `pad_value`, or `axis` are not scalar. framer)namerrrr rcSs"t|}|dur |dfS|dfS)NTF)rr )valZ val_staticrrrmaybe_constants zframe..maybe_constant)Zindices_or_sections)dtype)Zconstant_values)rN)rZ name_scopeZconvert_to_tensorr Zwith_rank_at_leastZassert_has_rankrrrangelennpsplititemZrankrZreshapesizer!maximumconcatZzerospadrZconstantrgcdZ strided_sliceZ zeros_likegather set_shape)"rrrrZ pad_valuerrZ result_shaperrZsignal_shape_is_staticZaxis_is_staticrZlength_samplesrZ signal_rankZnum_outer_dimensionsZnum_inner_dimensionsrZ pad_samplesZpaddingsZsubframe_length_Zsubframes_per_frameZsubframes_per_hopZ num_subframesZ slice_shapeZsubframe_shapeZ subframesZframe_selectorZsubframe_selectorselectorZmid_dimensionsframesrrrr7s;                      &r)FrrN)__doc__numpyr$Ztensorflow.python.frameworkrrrZtensorflow.python.opsrrZtensorflow.python.ops.signalrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportr rZadd_dispatch_supportrrrrrs