torchaudio

I/O 功能

音频输入/输出功能在 torchaudio.backend 模块中实现，但为了方便使用，以下函数在 torchaudio 模块中提供。有多种后端可供选择，您可以使用 set_audio_backend() 切换后端。

有关详细信息，请参阅 torchaudio.backend。

torchaudio.info(filepath: str, ...)

获取音频文件的元数据。有关详细信息，请参阅 torchaudio.backend。

torchaudio.load(filepath: str, ...)

将音频文件加载到 torch.Tensor 对象中。有关详细信息，请参阅 torchaudio.backend。

torchaudio.load_wav(filepath: str, ...)

加载音频文件到torch.Tensor，有关详细信息请参阅torchaudio.backend。

torchaudio.save(filepath: str, src: torch.Tensor, sample_rate: int, ...)

将 torch.Tensor 对象保存为音频格式。有关详细信息，请参阅 torchaudio.backend。

后端工具

torchaudio.list_audio_backends() → List[str]

列出可用的后端

Returns

可用后端的列表。

Return type

列表[字符串]

torchaudio.get_audio_backend() → Optional[str]

获取当前后端的名称

Returns

当前后端的名称，或者如果未分配后端则为 None。

Return type

Optional[str]

torchaudio.set_audio_backend(backend: Optional[str])

设置 I/O 操作的后端

Parameters

后端 (可选[字符串]) – 后端名称。 其中一个 "sox", "sox_io" 或 "soundfile" 根据系统的可用性 如果提供 None 当前后端未分配。