Package io.github.kirstenali.deepj.tensor

Class CrossEntropyDefaults

java.lang.Object

io.github.kirstenali.deepj.tensor.CrossEntropyDefaults

public final class CrossEntropyDefaults
extends Object

Unfused backend composition preserves the original matrix loss/gradient evaluation order.

Method Summary

Modifier and Type	Method	Description
static CrossEntropyResult	apply(TensorBackend backend, 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(TensorBackend backend,
 Tensor logits,
 int[] targets,
 boolean[] mask)