Tensorflow: binary mask of max values along tensor axis

Question

If I have a N-dimensional tensor, I would like to create another tensor (with the same shape) of values 0 and 1, where 1 is in the same position as the maximum element in origin