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