def audio(tag, tensor, sample_rate=44100):
tensor = make_np(tensor)
tensor = tensor.squeeze()
if abs(tensor).max() > 1:
print("warning: audio amplitude out of range, auto clipped.")
tensor = tensor.clip(-1, 1)
assert(tensor.ndim == 1), "input tensor should be 1 dimensional."
tensor_list = [int(32767.0 * x) for x in tensor]
import io
import wave
import struct
After Change
assert num_channels == 1 or num_channels == 2, "The second dimension should be 1 or 2."
with io.BytesIO() as fio:
soundfile.write(fio, tensor, samplerate=sample_rate, format="wav")
audio_string = fio.getvalue()
audio = Summary.Audio(sample_rate=sample_rate,
num_channels=num_channels,
