Type Alias SoftmaxOp<Input, Axis, OutputDType>

SoftmaxOp: ValidateDim<Axis, Input["__shape"]> extends DimensionError<string>
    ? never
    : StorageTransformation<
        "softmax",
        TensorStorage<
            OutputDType,
            Input["__shape"],
            ComputeStrides<Input["__shape"]>
            | Input["__strides"],
            SoftmaxOpLayout<Input["__layout"]>,
        >,
        readonly [Input],
    > & { __softmaxAxis: ValidateDim<Axis, Input["__shape"]> }