AVh'dZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZd Zed e j" dd Zy)zGeneral shape ops for frames.N) constant_op)ops) tensor_util) array_ops)math_ops)util_ops)dispatch) tf_exportctj|}tj|}tj|}|jjy|dg|jjdzzS|jj }d}||}|d|}||ddd} |r5|3||rt d| |z }n|||rJt d||z |z|z}|||gz| zS)z0Infers the shape of the return value of `frame`.Nr)rconstant_valueshapendimsas_listmax) signal frame_length frame_steppad_endaxis signal_shape num_frames frame_axisouter_dimensionsinner_dimensionss V/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/signal/shape_ops.py_infer_frame_shapers++L9,))*5*  # #D )$ \\  \ 6V\\''!+ ,,%%',*D!*!%4(!$%(,j,'qZK:567j  L$<[ j<'*4 CEj Z6 69I IIz signal.framec J tj|d||||g5tj|d}tj|d}tj|d}tj|d}|jj d|jj d|jj d|jj dt |||||}d }|tj|\} } ||\}} | rJ| rHtt| |}tj| ||dzg \} } }| j} nftj|}tj||}tj| |d|dz |z g\} } }tj| g} tj | }tj |}|rtj||j"}|jj d| |z }tj$d|||dz zz| z }tj&tj(|d g|j" tjd|ggtj(|d g|j" gd}tj*||| }tj|} | |} n>tj$t-j.d|j" d| |z |zz}|t1j2||\}}||z}||z}| |z}tj&| ||zg|gd}tj&| ||g|gd}tjtj4|tj6| ||}tjtj||j" |z|dg}tjtj||j" d|g}||z}tj| } tj|| j" }tj||g| j" } tjtj8|||tj&| | |gd}!|r|!j;||!cdddS#1swYyxYw)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 rc@tj|}||dfS|dfS)NTF)rr )val val_statics rmaybe_constantzframe..maybe_constants+--c2j#-#9j$ KU|Kr)indices_or_sections)dtype)constant_values)rN)r name_scopeconvert_to_tensorrwith_rank_at_leastassert_has_rankrrrangelennpsplititemrankrreshapesizer(maximumconcatzerospadrconstantrgcd strided_slice zeros_likegather set_shape)"rrrr pad_valuerr! result_shaper%rsignal_shape_is_staticaxis_is_staticrlength_samplesr signal_ranknum_outer_dimensionsnum_inner_dimensionsr pad_samplespaddingssubframe_length_subframes_per_framesubframes_per_hop num_subframes slice_shapesubframe_shape subframesframe_selectorsubframe_selectorselectormid_dimensionsframess" rr r 7sv ~~dGflJ&/&12u  " "6 9F((NKL&&z EJ  F 3D LL##A&&&q)$$Q'JJq!%flJ&*,LL,:,!(L()$/D.. 3|$ %d +d;=88 T4!8,<<>8(8%**,nNN6*k ^^K ( .d;D?? q+/D"89<;8(8 ((>*:;$>>*:;'' 6<<@ioo%%a(%_ 23j$$ \J*q.9 9N JLk!! ///3;;L;L M  ![!1 2 3 ///3;;L;L M#  h }}VXyIf__V,l#D)n##   q (:(: ; ~ ,; ;=j( \:(NOOQ&/9"o5"o5M""$4%2_%D$E$4$678:K%%'7(5'G'7'9:;=N!!)"9"9 $$\2#$&I&&z););< &?,N"))*,2D2DE  " 11H,,-=>,, 0 6 68**J +E1A1G1GIN  H48*NrasZ$3+3+*1+6J4 > MOnnr