По-видимому, самое внутреннее в этой документации использовалось для последнего измерения в массиве.В случае тензорного потока, однозначно, комментарии кода работают лучше, чем документация.Официальная документация по тензорному потоку все еще отстой.