mirror of
https://github.com/thxxx/VTS.git
synced 2026-06-25 03:14:06 +00:00
11 lines
408 B
Python
11 lines
408 B
Python
from jaxtyping import Bool, Float, Int
|
|
from torch import Tensor
|
|
|
|
AudioTensor = Float[Tensor, "batch audio audio_channel"]
|
|
AudioMaskTensor = Bool[Tensor, "batch audio"]
|
|
EncTensor = Float[Tensor, "batch codec channel"]
|
|
EncMaskTensor = Bool[Tensor, "batch codec"]
|
|
LengthTensor = Int[Tensor, "batch"]
|
|
LossTensor = Float[Tensor, ""]
|
|
TimeTensor = Float[Tensor, "batch"]
|
|
Batch = tuple[AudioTensor, AudioMaskTensor]
|