io.improbable.keanu.tensor

Interface Tensor<T>

All Known Subinterfaces:

BooleanTensor, DoubleTensor, IntegerTensor, NumberTensor<N,T>

All Known Implementing Classes:

GenericTensor, Nd4jDoubleTensor, Nd4jIntegerTensor, ScalarDoubleTensor, ScalarIntegerTensor, SimpleBooleanTensor

public interface Tensor<T>

Nested Class Summary

Nested Classes

Modifier and Type	Interface and Description
static interface	Tensor.FlattenedView<T>

Field Summary

Fields

Modifier and Type	Field and Description
static long[]	ONE_BY_ONE_SHAPE
static long[]	SCALAR_SHAPE
static long[]	SCALAR_STRIDE

Method Summary

Modifier and Type	Method and Description
T[]	asFlatArray()
double[]	asFlatDoubleArray()
int[]	asFlatIntegerArray()
default java.util.List<T>	asFlatList()
static <DATA,TENSOR extends Tensor<DATA>> TENSOR	create(DATA[] data, long[] shape)
static <DATA,TENSOR extends Tensor<DATA>> TENSOR	createFilled(DATA data, long[] shape)
Tensor<T>	duplicate()
BooleanTensor	elementwiseEquals(T value)
default BooleanTensor	elementwiseEquals(Tensor that)
static BooleanTensor	elementwiseEquals(Tensor a, Tensor b)
Tensor.FlattenedView<T>	getFlattenedView()
long	getLength()
int	getRank()
long[]	getShape()
default T	getValue(int... index)
T	getValue(long... index)
default boolean	hasSameShapeAs(long[] shape)
default boolean	hasSameShapeAs(Tensor that)
default boolean	isLengthOne()
default boolean	isMatrix()
default boolean	isScalar()
boolean	isShapePlaceholder()
default boolean	isVector() Returns true if the tensor is a vector.
Tensor<T>	reshape(long... newShape)
T	scalar()
static <DATA,TENSOR extends Tensor<DATA>> TENSOR	scalar(DATA data)
Tensor<T>	setValue(T value, long... index)
Tensor<T>	slice(int dimension, long index)

Field Detail

SCALAR_SHAPE

static final long[] SCALAR_SHAPE

SCALAR_STRIDE

static final long[] SCALAR_STRIDE

ONE_BY_ONE_SHAPE

static final long[] ONE_BY_ONE_SHAPE

Method Detail

scalar

static <DATA,TENSOR extends Tensor<DATA>> TENSOR scalar(DATA data)

createFilled

static <DATA,TENSOR extends Tensor<DATA>> TENSOR createFilled(DATA data,
                                                              long[] shape)

create

static <DATA,TENSOR extends Tensor<DATA>> TENSOR create(DATA[] data,
                                                        long[] shape)

elementwiseEquals

static BooleanTensor elementwiseEquals(Tensor a,
                                       Tensor b)

getRank

int getRank()

getShape

long[] getShape()

getLength

long getLength()

isShapePlaceholder

boolean isShapePlaceholder()

getValue

T getValue(long... index)

getValue

default T getValue(int... index)

setValue

Tensor<T> setValue(T value,
                   long... index)

scalar

T scalar()

duplicate

Tensor<T> duplicate()

slice

Tensor<T> slice(int dimension,
                long index)

asFlatDoubleArray

double[] asFlatDoubleArray()

asFlatIntegerArray

int[] asFlatIntegerArray()

asFlatArray

T[] asFlatArray()

reshape

Tensor<T> reshape(long... newShape)

getFlattenedView

Tensor.FlattenedView<T> getFlattenedView()

asFlatList

default java.util.List<T> asFlatList()

isLengthOne

default boolean isLengthOne()

isScalar

default boolean isScalar()

isVector

default boolean isVector()

Returns true if the tensor is a vector. A vector being a 1xn or a nx1 tensor.

(1, 2, 3) is a 1x3 vector.

(1) (2) (3) is a 3x1 vector.

Returns:

true if the tensor is a vector

isMatrix

default boolean isMatrix()

hasSameShapeAs

default boolean hasSameShapeAs(Tensor that)

hasSameShapeAs

default boolean hasSameShapeAs(long[] shape)

elementwiseEquals

default BooleanTensor elementwiseEquals(Tensor that)

elementwiseEquals

BooleanTensor elementwiseEquals(T value)