softmax(axis:)
Adds a softmax along the given axis operation to the current graph.
Declaration
func softmax(axis: Int) -> BNNSGraph.Builder.Tensor<T>Parameters
- axis:
The axis over which the operation computes softmax.
Adds a softmax along the given axis operation to the current graph.
func softmax(axis: Int) -> BNNSGraph.Builder.Tensor<T>The axis over which the operation computes softmax.