o siT@s,ddlZddlmZeddefddZdS) N)script_if_tracing(th_dbcCsNdt|}t|jddddg}tj||dddd}|||kS) aCompute energy-based VAD from a magnitude spectrogram (or equivalent). Args: mag_spec (torch.Tensor): the spectrogram to perform VAD on. Expected shape (batch, *, freq, time). The VAD mask will be computed independently for all the leading dimensions until the last two. Independent of the ordering of the last two dimensions. th_db (int): The threshold in dB from which a TF-bin is considered silent. Returns: :class:`torch.BoolTensor`, the VAD mask. Examples >>> import torch >>> mag_spec = torch.abs(torch.randn(10, 2, 65, 16)) >>> batch_src_mask = ebased_vad(mag_spec) NT)keepdimr)torchlog10listshapemaxview)mag_specrlog_magto_view max_log_magrD/home/ubuntu/.local/lib/python3.10/site-packages/asteroid/dsp/vad.py ebased_vads r)r)r utils.torch_utilsrintrrrrrs