Function to convert an index from the vocabulary into tensor

index_from_tensor.py

# Define a function to convert tensor into index in vocabulary
def indexFromTensor(target):
    '''
    Function returns tensor containing target index given tensor representing target word
    '''
    top_n, top_i = target.topk(1)
    target_index = top_i[0].item()
    
    target_index_tensor = torch.zeros((1), dtype = torch.long)
    target_index_tensor[0] = target_index
    return target_index_tensor