Package io.github.kirstenali.deepj.tensor

Class CpuNdElementwise

java.lang.Object

io.github.kirstenali.deepj.tensor.CpuNdElementwise

public final class CpuNdElementwise
extends Object

CPU logical-layout kernels with real compact storage and FLOAT32 arithmetic.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	CpuNdElementwise.Binary
static interface	CpuNdElementwise.Unary

Method Summary

Modifier and Type	Method	Description
static Tensor	binary(Tensor left, Tensor right, CpuNdElementwise.Binary operation)
static void	binaryInto(Tensor left, Tensor right, Tensor output, CpuNdElementwise.Binary operation)
static Tensor	cast(Tensor input, DType dtype)
static boolean	matrix(Tensor input)
static boolean	sameMatrices(Tensor left, Tensor right)
static Tensor	unary(Tensor input, CpuNdElementwise.Unary operation)
static void	unaryInto(Tensor input, Tensor output, CpuNdElementwise.Unary operation)

Methods inherited from class java.lang.Object

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

Method Details

matrix

public static boolean matrix(Tensor input)

sameMatrices

public static boolean sameMatrices(Tensor left,
 Tensor right)

unary

public static Tensor unary(Tensor input,
 CpuNdElementwise.Unary operation)

unaryInto

public static void unaryInto(Tensor input,
 Tensor output,
 CpuNdElementwise.Unary operation)

binary

public static Tensor binary(Tensor left,
 Tensor right,
 CpuNdElementwise.Binary operation)

binaryInto

public static void binaryInto(Tensor left,
 Tensor right,
 Tensor output,
 CpuNdElementwise.Binary operation)

cast

public static Tensor cast(Tensor input,
 DType dtype)