Package 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 Description
    TensorFactory​(java.lang.Class tensorType)  

  • Method Summary

    Modifier and Type Method 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 Details

    • TensorFactory

      public TensorFactory​(java.lang.Class tensorType)

  • Method Details

    • 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()