o 2hN@sFdZddlZddlmZddlmZddlmZ    d ddZ dS) aAudio summaries and TensorFlow operations to create them, V2 versions. An audio summary stores a rank-2 string tensor of shape `[k, 2]`, where `k` is the number of audio clips recorded in the summary. Each row of the tensor is a pair `[encoded_audio, label]`, where `encoded_audio` is a binary string whose encoding is specified in the summary metadata, and `label` is a UTF-8 encoded Markdown string describing the audio clip. N)tf2)metadata)lazy_tensor_creatorc sttdddurddlm|durd}|dkr td|tjd|tjdd}|g}ttj j d dp=tj j } | |d |d  \} } t j fd d } tj j| | ||dWdS1shwYdS)aWrite an audio summary. Arguments: name: A name for this summary. The summary tag used for TensorBoard will be this name prefixed by any active name scopes. data: A `Tensor` representing audio data with shape `[k, t, c]`, where `k` is the number of audio clips, `t` is the number of frames, and `c` is the number of channels. Elements should be floating-point values in `[-1.0, 1.0]`. Any of the dimensions may be statically unknown (i.e., `None`). sample_rate: An `int` or rank-0 `int32` `Tensor` that represents the sample rate, in Hz. Must be positive. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. max_outputs: Optional `int` or rank-0 integer `Tensor`. At most this many audio clips will be emitted at each step. When more than `max_outputs` many clips are provided, the first `max_outputs` many clips will be used and the rest silently discarded. encoding: Optional constant `str` for the desired encoding. Only "wav" is currently supported, but this is not guaranteed to remain the default, so if you want "wav" in particular, set this explicitly. description: Optional long-form description for this summary, as a constant `str`. Markdown is supported. Defaults to empty. Returns: True on success, or false if no summary was emitted because no default summary writer was available. Raises: ValueError: if a default writer exists, but no step was provided and `tf.summary.experimental.get_step()` is None. audioNr) gen_audio_opswavzUnknown encoding: %rWAV) display_name descriptionencoding summary_scope audio_summary)valuescstjdtjd}tjjd}tj||tjddt tj dddkfddd dt d gtj |ddd }tj t |gd S) Nr) sample_rateencode_each_audio)dtypename)inputrcsSNr encoded_audior_/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/plugins/audio/summary_v2.pyssz,audio..lazy_tensor..cSstgtjSr)tfconstantstringrrrrrts)a)r debugging assert_rankassert_non_negative functoolspartial encode_wavmap_fnrcondshapetile transposestack) limited_audio encode_fnlimited_labels audio_opsdata max_outputsrrr lazy_tensorbs&   zaudio..lazy_tensor)tagtensorstepr)getattrrtensorflow.python.opsr ValueErrorrcreate_summary_metadataEncodingValuesummary experimentalr rLazyTensorCreatorwrite) rr2rr7r3r r summary_metadatainputsr r5_r4rr0rr s. *    $r)NrNN) __doc__r$tensorboard.compatrrtensorboard.plugins.audiortensorboard.utilrrrrrrs