U %d@sPddlmZmZmZmZddlZddlmZddlmZ e ddrTddl Z ddl Z ddd d d gZdeeeeeeefd ddZe ddeeeeefdddZe ddeeeeefdddZe ddeeeeefddd Ze ddeeeeefddd Ze ddeeeeefddd ZdS))AnyCallableIterableTupleN)Tensor) module_utilskaldi_ionumpyread_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_arrr7/tmp/pip-unpacked-wheel-lbdmvq91/torchaudio/kaldi_io.py _convert_method_output_to_tensors r)rrcCst|tjddS)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) } T)r)rrr rrrrr -scCs t|tjS)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) } )rrr rrrrr AscCs t|tjS)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) } )rrr rrrrr RscCs t|tjS)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) } )rrr rrrrr cscCs t|tjS)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) } )rrrrrrrrts)F)typingrrrrrrZtorchaudio._internalrZ _mod_utilsZis_module_availablerr r__all__boolstrrZrequires_moduler r r r rrrrrs: