VGGishBundle.VGGish

class torchaudio.prototype.pipelines.VGGishBundle.VGGish

VGGish 模型的實作 [Hershey et al., 2017]。

方法

forward

VGGish.forward(input: Tensor) → Tensor

參數:

input (torch.Tensor) – 頻譜圖批次，形狀為 (n_example, 1, n_frame, 64)。

回傳:

模型輸出，形狀為 (n_example, 128)。

回傳類型:

torch.Tensor