Если вы уже делаете проект или выполняли задание на машинный перевод, наверняка вы путались в том, за что отвечают размерности теноров. HarvardNLP нашёл этому решение - namedTensor и рекомендует всем его использовать
http://nlp.seas.harvard.edu/NamedTensor.html
http://nlp.seas.harvard.edu/NamedTensor.html