使用tf.data.map将音频文件分割成1秒的音频张量块？

import tensorflow as tf
import librosa

def load_and_segment_audio(file_path):
    audio, _ = librosa.load(file_path, sr=None)  # 加载音频文件
    segment_length = 1 * audio.shape[0]  # 计算1秒的音频长度
    segments = tf.signal.frame(audio, frame_length=segment_length, frame_step=segment_length, pad_end=True)  # 分割音频
    return segments

file_list = ['audio1.wav', 'audio2.wav', 'audio3.wav']  # 音频文件列表
dataset = tf.data.Dataset.from_tensor_slices(file_list)

dataset = dataset.map(load_and_segment_audio)