distribution-in-tensorflow