softmax

fun DagBuilder.softmax(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for softmax.