ag_softmax: Softmax activation (column-wise)

View source: R/autograd.R

ag_softmaxR Documentation

Softmax activation (column-wise)

Description

Applies numerically stable softmax along rows so that each column (one sample) sums to 1.

Usage

ag_softmax(x)

Arguments

x

ag_tensor of shape [classes, batch_size]

Value

ag_tensor of the same shape as x


ggmlR documentation built on July 14, 2026, 1:08 a.m.