Package io.github.kirstenali.deepj.tensor

Class CpuRowAccess

java.lang.Object

io.github.kirstenali.deepj.tensor.CpuRowAccess

public final class CpuRowAccess
extends Object

Checked CPU row kernels; the Tensor facade owns autograd mutation fences as for other backends.

Method Summary

Modifier and Type	Method	Description
static void	assign(Tensor target, int row, Tensor source, int sourceRow)
static Tensor	sample(Tensor input, int count, Random random)
static Tensor	select(Tensor input, int[] rows, int columns)

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[] rows,
 int columns)

assign

public static void assign(Tensor target,
 int row,
 Tensor source,
 int sourceRow)

sample

public static Tensor sample(Tensor input,
 int count,
 Random random)