import torch


def pitchVocabularyFmt(X, vocab_col):
    """
    Produces the tensors for training with a pitch vocabulary encoding.
    """
    pitch = torch.tensor(X[:, vocab_col], dtype=torch.long)
    score_feats = torch.cat([torch.tensor(X[:, :vocab_col], dtype=torch.float),
                            torch.tensor(X[:, vocab_col + 1:], dtype=torch.float)], dim=1)
    return pitch, score_feats