:jj VddlmZmZmZmZddlZddlmZddlmZ ddl m Z e jdrddl ZgdZ dded ed ed eeeeffd Ze e j&d dded eeeeffdZe e j&d dded eeeeffdZe e j&d dded eeeeffdZe e j&d dded eeeeffdZe e j&d dded eeeeffdZy))AnyCallableIterableTupleN)Tensor) module_utils)dropping_supportnumpy)read_vec_int_arkread_vec_flt_scpread_vec_flt_ark read_mat_scp read_mat_ark file_or_fdfnconvert_contiguousreturnc#K||D]5\}}|rtj|}|tj|f7yw)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)npascontiguousarraytorch from_numpy)rrrkeynp_arrs h/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/torchaudio/kaldi_io.py _convert_method_output_to_tensorrsH*~, V ))&1F5##F+++,sAAkaldi_ioc:ddl}t||jdS)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 rrs rr r -s  ,J8Q8Qfj kkc6ddl}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 rs rr r E  +J8Q8Q RRr c6ddl}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 rs rr r Zr"r c6ddl}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)rrrrs rrro  +J8M8M NNr c6ddl}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)rrrrs rrrr%r )F)typingrrrrrrtorchaudio._internalr _mod_utils!torchaudio._internal.module_utilsr is_module_availabler r__all__boolstrrrequires_moduler r r rrr rr1s21 ;>!:!!'* ?D,,!,7;, eCK !,*J0ll%V 2D)El1l,J0SS%V 2D)ES1S&J0SS%V 2D)ES1S&J0OSOXeCK.@%AO1O&J0OSOXeCK.@%AO1Or