Package io.github.kirstenali.deepj.tensor

Class CpuNdCrossEntropy

java.lang.Object

io.github.kirstenali.deepj.tensor.CpuNdCrossEntropy

public final class CpuNdCrossEntropy
extends Object

Stable final-axis loss and FLOAT32 gradient, including strided/compact logits and response masks.

Method Summary

Modifier and Type	Method	Description
static CrossEntropyResult	apply(Tensor logits, int[] targets, boolean[] mask)
static float	loss(Tensor logits, int[] targets, boolean[] mask)

Methods inherited from class java.lang.Object

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

Method Details

apply

public static CrossEntropyResult apply(Tensor logits,
 int[] targets,
 boolean[] mask)

loss

public static float loss(Tensor logits,
 int[] targets,
 boolean[] mask)