import torch


def init_tdl_zeros(model, batch_size, device):

    """
    Initialize TDLs with zeros.

    :param model: NARX model
    :param batch_size: size of batch
    :param device: device type
    :return: input TDL, output TDL
    """

    # input tap-delay
    itdl = torch.zeros((batch_size, model.input_delay_size, model.input_size), dtype=torch.float64).to(device)

    # output tap-delay
    otdl = torch.zeros((batch_size, model.output_delay_size, model.output_size), dtype=torch.float64).to(device)

    return itdl, otdl