replicated softmax