io.improbable.keanu.tensor.generic

Class GenericTensor<T>

java.lang.Object

io.improbable.keanu.tensor.jvm.JVMTensor<T,GenericTensor<T>,GenericBuffer.PrimitiveGenericWrapper<T>>

io.improbable.keanu.tensor.generic.GenericTensor<T>

All Implemented Interfaces:

Tensor<T,GenericTensor<T>>

public class GenericTensor<T>
extends JVMTensor<T,GenericTensor<T>,GenericBuffer.PrimitiveGenericWrapper<T>>
implements Tensor<T,GenericTensor<T>>

Nested Class Summary

Nested classes/interfaces inherited from interface io.improbable.keanu.tensor.Tensor

Tensor.FlattenedView<N>

Field Summary

Fields inherited from class io.improbable.keanu.tensor.jvm.JVMTensor

buffer, shape, stride

Fields inherited from interface io.improbable.keanu.tensor.Tensor

ONE_BY_ONE_SHAPE, SCALAR_SHAPE, SCALAR_STRIDE

Method Summary

Modifier and Type	Method and Description
<R> GenericTensor<R>	apply(Tensor<T,?> right, java.util.function.BiFunction<T,T,R> op)
T[]	asFlatArray()
protected GenericTensor<T>	create(GenericBuffer.PrimitiveGenericWrapper<T> buffer, long[] shape, long[] stride)
static <T> GenericTensor<T>	create(T... data)
static <T> GenericTensor<T>	create(T[] data, long[] shape)
static <T> GenericTensor<T>	createFilled(T data, long[] shape)
GenericTensor<T>	duplicate()
BooleanTensor	elementwiseEquals(T value)
protected JVMBuffer.ArrayWrapperFactory<T,GenericBuffer.PrimitiveGenericWrapper<T>>	getFactory()
Tensor.FlattenedView<T>	getFlattenedView()
static <T> GenericTensor<T>	scalar(T data)
protected GenericTensor<T>	set(GenericBuffer.PrimitiveGenericWrapper<T> buffer, long[] shape, long[] stride)
GenericTensor<T>	take(long... index)

Methods inherited from class io.improbable.keanu.tensor.jvm.JVMTensor

argCompare, argCompare, argCompare, argCompare, broadcast, broadcastableBinaryOpWithAutoBroadcast, concat, concatOnDimensionZero, diag, diag, equals, get, getLength, getRank, getShape, getStride, hashCode, isApply, permute, permute, reshape, slice, slice, slice, split, split, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface io.improbable.keanu.tensor.Tensor

asFlatList, broadcast, create, diag, elementwiseEquals, elementwiseEquals, expandDims, get, getLength, getRank, getShape, getStride, getValue, hasSameShapeAs, hasSameShapeAs, isLengthOne, isMatrix, isScalar, isVector, moveAxis, permute, reshape, scalar, setValue, slice, slice, slice, sliceAlongDimension, split, squeeze, swapAxis, transpose

Method Detail

createFilled

public static <T> GenericTensor<T> createFilled(T data,
                                                long[] shape)

Specified by:

createFilled in interface Tensor<T,GenericTensor<T>>

create

public static <T> GenericTensor<T> create(T... data)

create

public static <T> GenericTensor<T> create(T[] data,
                                          long[] shape)

scalar

public static <T> GenericTensor<T> scalar(T data)

Specified by:

scalar in interface Tensor<T,GenericTensor<T>>

duplicate

public GenericTensor<T> duplicate()

Specified by:

duplicate in interface Tensor<T,GenericTensor<T>>

getFlattenedView

public Tensor.FlattenedView<T> getFlattenedView()

Specified by:

getFlattenedView in interface Tensor<T,GenericTensor<T>>

elementwiseEquals

public BooleanTensor elementwiseEquals(T value)

Specified by:

elementwiseEquals in interface Tensor<T,GenericTensor<T>>

asFlatArray

public T[] asFlatArray()

Specified by:

asFlatArray in interface Tensor<T,GenericTensor<T>>

create

protected GenericTensor<T> create(GenericBuffer.PrimitiveGenericWrapper<T> buffer,
                                  long[] shape,
                                  long[] stride)

Specified by:

create in class JVMTensor<T,GenericTensor<T>,GenericBuffer.PrimitiveGenericWrapper<T>>

set

protected GenericTensor<T> set(GenericBuffer.PrimitiveGenericWrapper<T> buffer,
                               long[] shape,
                               long[] stride)

Specified by:

set in class JVMTensor<T,GenericTensor<T>,GenericBuffer.PrimitiveGenericWrapper<T>>

getFactory

protected JVMBuffer.ArrayWrapperFactory<T,GenericBuffer.PrimitiveGenericWrapper<T>> getFactory()

Specified by:

getFactory in class JVMTensor<T,GenericTensor<T>,GenericBuffer.PrimitiveGenericWrapper<T>>

take

public GenericTensor<T> take(long... index)

Specified by:

take in interface Tensor<T,GenericTensor<T>>

apply

public <R> GenericTensor<R> apply(Tensor<T,?> right,
                                  java.util.function.BiFunction<T,T,R> op)