Package io.github.kirstenali.deepj.autograd

Class AxisReductionOps

java.lang.Object

io.github.kirstenali.deepj.autograd.AxisReductionOps

public final class AxisReductionOps
extends Object

Method Summary

Modifier and Type	Method	Description
static Tensor	maximum(Tensor input, int[] axes, boolean keep)
static Tensor	mean(Tensor input, int[] axes, boolean keep)
static Tensor	softmax(Tensor input, int axis)
static Tensor	sum(Tensor input, int[] axes, boolean keep)
static Tensor	variance(Tensor input, int[] axes, boolean keep, int correction)

Methods inherited from class java.lang.Object

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

Method Details

sum

public static Tensor sum(Tensor input,
 int[] axes,
 boolean keep)

softmax

public static Tensor softmax(Tensor input,
 int axis)

mean

public static Tensor mean(Tensor input,
 int[] axes,
 boolean keep)

variance

public static Tensor variance(Tensor input,
 int[] axes,
 boolean keep,
 int correction)

maximum

public static Tensor maximum(Tensor input,
 int[] axes,
 boolean keep)