docs.unity3d.com
Search Results for

    Show / Hide Table of Contents

    Method Softmax

    Softmax(TensorFloat, int)

    Computes an output tensor by applying the Softmax activation function along an axis: f(X, axis) = exp(X) / ReduceSum(exp(X), axis).

    Declaration
    public TensorFloat Softmax(TensorFloat X, int axis = -1)
    Parameters
    Type Name Description
    TensorFloat X

    The input tensor.

    int axis

    The axis along which to apply the Softmax activation function. The default value is -1.

    Returns
    Type Description
    TensorFloat

    The computed output tensor.

    In This Article
    Back to top
    Copyright © 2024 Unity Technologies — Trademarks and terms of use
    • Legal
    • Privacy Policy
    • Cookie Policy
    • Do Not Sell or Share My Personal Information
    • Your Privacy Choices (Cookie Settings)