Попробуйте это
import tensorflow as tf
from tensorflow import distributions as dist
Я не думаю, что вы можете использовать псевдонимы импорта для других операторов импорта в python.Я не слишком уверен в этом, но я думаю, что это проблема.
К вашему сведению, я проверял это на Python 3.5.2 и Tensorflow 1.8.0