SoftmaxOptions

import { SoftmaxOptions } from "destack:tensor";

Options for softmax-like operations.

export struct SoftmaxOptions<T, const Rank: usize> {
    axis: Axis;
    output: TensorOutputOptions<T, Rank>;
}
language/library/src/tensor/neural.ds:10:16

Members

axis: Axis

Axis to normalize over.

output: TensorOutputOptions<T, Rank>

Output options.