a hj@s`ddlmZmZmZmZddlZddlmZddlmZ ddl m Z e drVddl ZgdZdeeeeeeefd d d Ze e d deeeeefd ddZe e d deeeeefd ddZe e d deeeeefd ddZe e d deeeeefd ddZe e d deeeeefd ddZdS))AnyCallableIterableTupleN)Tensor) module_utils)dropping_supportnumpy)read_vec_int_arkread_vec_flt_scpread_vec_flt_ark read_mat_scp read_mat_arkF) file_or_fdfnconvert_contiguousreturnccs4||D]&\}}|rt|}|t|fVqdS)aTakes a method invokes it. The output is converted to a tensor. Args: file_or_fd (str/FileDescriptor): File name or file descriptor fn (Callable): Function that has the signature (file name/descriptor) and converts it to Iterable[Tuple[str, Tensor]]. convert_contiguous (bool, optional): Determines whether the array should be converted into a contiguous layout. (Default: ``False``) Returns: Iterable[Tuple[str, Tensor]]: The string is the key and the tensor is vec/mat N)npZascontiguousarraytorchZ from_numpy)rrrkeyZnp_arrrA/var/www/auris/lib/python3.9/site-packages/torchaudio/kaldi_io.py _convert_method_output_to_tensors rkaldi_io)rrcCsddl}t||jddS)aCreate generator of (key,vector) tuples, which reads from the ark file/stream. Args: file_or_fd (str/FileDescriptor): ark, gzipped ark, pipe or opened file descriptor Returns: Iterable[Tuple[str, Tensor]]: The string is the key and the tensor is the vector read from file Example >>> # read ark to a 'dictionary' >>> d = { u:d for u,d in torchaudio.kaldi_io.read_vec_int_ark(file) } rNT)r)rrr rrrrrr -sr cCsddl}t||jS)aCreate generator of (key,vector) tuples, read according to Kaldi scp. Args: file_or_fd (str/FileDescriptor): scp, gzipped scp, pipe or opened file descriptor Returns: Iterable[Tuple[str, Tensor]]: The string is the key and the tensor is the vector read from file Example >>> # read scp to a 'dictionary' >>> # d = { u:d for u,d in torchaudio.kaldi_io.read_vec_flt_scp(file) } rN)rrr rrrrr Esr cCsddl}t||jS)aCreate generator of (key,vector) tuples, which reads from the ark file/stream. Args: file_or_fd (str/FileDescriptor): ark, gzipped ark, pipe or opened file descriptor Returns: Iterable[Tuple[str, Tensor]]: The string is the key and the tensor is the vector read from file Example >>> # read ark to a 'dictionary' >>> d = { u:d for u,d in torchaudio.kaldi_io.read_vec_flt_ark(file) } rN)rrr rrrrr Zsr cCsddl}t||jS)aCreate generator of (key,matrix) tuples, read according to Kaldi scp. Args: file_or_fd (str/FileDescriptor): scp, gzipped scp, pipe or opened file descriptor Returns: Iterable[Tuple[str, Tensor]]: The string is the key and the tensor is the matrix read from file Example >>> # read scp to a 'dictionary' >>> d = { u:d for u,d in torchaudio.kaldi_io.read_mat_scp(file) } rN)rrr rrrrr osr cCsddl}t||jS)aCreate generator of (key,matrix) tuples, which reads from the ark file/stream. Args: file_or_fd (str/FileDescriptor): ark, gzipped ark, pipe or opened file descriptor Returns: Iterable[Tuple[str, Tensor]]: The string is the key and the tensor is the matrix read from file Example >>> # read ark to a 'dictionary' >>> d = { u:d for u,d in torchaudio.kaldi_io.read_mat_ark(file) } rN)rrrrrrrrsr)F)typingrrrrrrZtorchaudio._internalrZ _mod_utilsZ!torchaudio._internal.module_utilsrZis_module_availabler r__all__boolstrrZrequires_moduler r r r rrrrrs6