Class CrossEntropyChecks

java.lang.Object
io.github.kirstenali.deepj.tensor.CrossEntropyChecks

public final class CrossEntropyChecks extends Object
Class scores occupy the final axis; every leading position has one integer target.
  • Method Details

    • classes

      public static int classes(Tensor logits)
    • positions

      public static int positions(Tensor logits)
    • requireTargets

      public static void requireTargets(Tensor logits, int[] targets)
    • included

      public static int included(boolean[] mask, int positions)