Package io.github.kirstenali.deepj.tensor

Class CpuNdIndexing

java.lang.Object

io.github.kirstenali.deepj.tensor.CpuNdIndexing

public final class CpuNdIndexing
extends Object

Logical ND index selection and deterministic repeated-index accumulation.

Method Summary

Modifier and Type	Method	Description
static void	add(Tensor target, int axis, int[] indices, Tensor updates)
static void	addMatrixRows(Tensor target, int[] indices, Tensor updates)
static Tensor	gradient(Tensor gradient, int[] shape, int axis, int[] indices)
static Tensor	select(Tensor input, int axis, int[] indices)

Methods inherited from class java.lang.Object

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

Method Details

select

public static Tensor select(Tensor input,
 int axis,
 int[] indices)

add

public static void add(Tensor target,
 int axis,
 int[] indices,
 Tensor updates)

addMatrixRows

public static void addMatrixRows(Tensor target,
 int[] indices,
 Tensor updates)

gradient

public static Tensor gradient(Tensor gradient,
 int[] shape,
 int axis,
 int[] indices)