P/j ddlmZmZmZmZddlZddlmZddlmZ e jdrddl Z gdZ ddeded ed eeeeffd Ze j"d dded eeeeffd Ze j"d dded eeeeffdZe j"d dded eeeeffdZe j"d dded eeeeffdZe j"d dded eeeeffdZy))AnyCallableIterableTupleN)Tensor) module_utilsnumpy)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 V/root/repos/pocs/video-evals/.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 C +J8Q8Q RRrc6ddl}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 Wr!rc6ddl}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 rs rr r k +J8M8M NNrc6ddl}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_utilsis_module_availabler r__all__boolstrrrequires_moduler r r r rrrr/s21 ;!:!!'* ?D,,!,7;, eCK !,*J0ll%V 2D)El1l,J0SS%V 2D)ES1S&J0SS%V 2D)ES1S&J0OSOXeCK.@%AO1O&J0OSOXeCK.@%AO1Or