U VcN@s>dZddlZddlmZddlmZddlmZd 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 sttdddkr ddlm|dkr,d}|dkr@td|tjd|tjdd}|g}ttj j d dpztj j } | |d |d >\} } t j fd d } tj j| | ||dW5QRSQRXdS)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_opsZwavzUnknown encoding: %rZWAV)Z display_name descriptionencoding summary_scopeZ audio_summary)valuescstjdtjd}tjjd}tj||tjddt tj dddkfddd dt d gtj |ddd }tj t |gd S) Nr) sample_rateZencode_each_audio)Zdtypename)inputrcsSNrZ encoded_audiorH/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/plugins/audio/summary_v2.pysz,audio..lazy_tensor..cSstgtjSr)tfZconstantstringrrrrrtr)a)r debuggingZ assert_rankZassert_non_negative functoolspartialZ encode_wavZmap_fnrZcondshapeZtileZ transposestack)Z limited_audioZ encode_fnZlimited_labelsZ audio_opsdata max_outputsr rr lazy_tensorbs(   zaudio..lazy_tensor)tagZtensorstepr)getattrrZtensorflow.python.opsr ValueErrorrZcreate_summary_metadataEncodingValuesummaryZ experimentalr rZLazyTensorCreatorwrite) r r r r$r!r rZsummary_metadatainputsr r#_r"rrrr s2*     r)NrNN) __doc__rZtensorboard.compatrrZtensorboard.plugins.audiorZtensorboard.utilrrrrrrs