added softmax with different temperatures