deepboof.misc

Class TensorFactory<T extends Tensor>

java.lang.Object

deepboof.misc.TensorFactory<T>

public class TensorFactory<T extends Tensor>
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
TensorFactory(java.lang.Class tensorType)

Method Summary

Modifier and Type	Method and Description
T	create(int... shape)
java.lang.Class<T>	getTensorType()
T	random(java.util.Random rand, boolean subTensor, double min, double max, int... shape)
T	random(java.util.Random rand, boolean subTensor, int... shape)
T	randomM(java.util.Random rand, boolean subTensor, int minibatch, int[] shape) Creates a random tensor with the specified shape and values from -1 to 1

Methods inherited from class java.lang.Object

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

Constructor Detail

TensorFactory

public TensorFactory(java.lang.Class tensorType)

Method Detail

create

public T create(int... shape)

randomM

public T randomM(java.util.Random rand,
                 boolean subTensor,
                 int minibatch,
                 int[] shape)

Creates a random tensor with the specified shape and values from -1 to 1

Parameters:

rand - Random number generator

subTensor - Should it be a sub-tensor or not?

minibatch - Number of mini-batches

shape - Shape of the tensor, without minibatch

Returns:

The random tensor

random

public T random(java.util.Random rand,
                boolean subTensor,
                int... shape)

random

public T random(java.util.Random rand,
                boolean subTensor,
                double min,
                double max,
                int... shape)

getTensorType

public java.lang.Class<T> getTensorType()