def parse_dt(dt, config):
    (fea, _, _, classes, _, target_class) = dt
    pos_fea = fea[:, :config.k_shot, :, 0, 0]
    neg_fea = fea[:, config.k_shot:, :, 0, 0]
    neg_shape = tf.shape(neg_fea)
    neg_fea = tf.reshape(neg_fea, [neg_shape[0], (- 1), neg_shape[(- 1)]])
    pos_classes = classes[:, :config.k_shot]
    neg_classes = classes[:, config.k_shot:]
    neg_classes = tf.reshape(neg_classes, [neg_shape[0], (- 1)])
    return (pos_fea, neg_fea, pos_classes, neg_classes, target_class)
def parse_dt(dt, config):
    (fea, _, _, classes, _, target_class) = dt
    pos_fea = fea[:, :config.k_shot, :, 0, 0]
    neg_fea = fea[:, config.k_shot:, :, 0, 0]
    neg_shape = tf.shape(neg_fea)
    neg_fea = tf.reshape(neg_fea, [neg_shape[0], (- 1), neg_shape[(- 1)]])
    pos_classes = classes[:, :config.k_shot]
    neg_classes = classes[:, config.k_shot:]
    neg_classes = tf.reshape(neg_classes, [neg_shape[0], (- 1)])
    return (pos_fea, neg_fea, pos_classes, neg_classes, target_class)