Package io.github.kirstenali.deepj.tensor

Class CpuNdSoftmax

java.lang.Object

io.github.kirstenali.deepj.tensor.CpuNdSoftmax

public final class CpuNdSoftmax
extends Object

Stable logical-axis softmax without an intermediate compact-dtype exponential tensor.

Method Summary

Modifier and Type	Method	Description
static Tensor	apply(Tensor input, int axis)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

apply

public static Tensor apply(Tensor input,
 int axis)