docs.unity3d.com
Search Results for

    Show / Hide Table of Contents

    Method Softmax

    Softmax(TensorFloat, 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 override void Softmax(TensorFloat X, TensorFloat O, int axis)
    Parameters
    Type Name Description
    TensorFloat X

    The input tensor.

    TensorFloat O

    The output tensor to be computed and filled.

    int axis

    The axis along which to apply the Softmax activation function.

    Overrides
    CPUBackend.Softmax(TensorFloat, TensorFloat, int)
    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)