Packages

sealed trait Tensor extends AnyRef

Self Type
Tensor
Source
Tensors.scala
Note

There are three kinds of Tensor.

  • InlineTensor and TransformedTensor are like a @inline def, which can be merged into a larger kernel and will be recalculated whenever a slow action is performed.
  • NonInlineTensor is like a @noinline def, which is never merged into a larger kernel and will be recalculated whenever a slow action is performed.
  • CachedTensor is like a lazy val, which has an associated data buffer on the compute device and will be calculated only once even when slow actions are performed more than once.
Linear Supertypes
AnyRef, Any
Known Subclasses
CachedTensor, FillTensor, InlineTensor, NonInlineTensor, TransformedTensor
Type Hierarchy
Ordering
  1. Grouped
  2. Alphabetic
  3. By Inheritance
Inherited
  1. Tensor
  2. AnyRef
  3. Any
Implicitly
  1. by any2stringadd
  2. by StringFormat
  3. by Ensuring
  4. by ArrowAssoc
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Abstract Value Members

  1. abstract val closure: (trees)#FloatTerm
    Attributes
    protected
  2. abstract def nonInline: NonInlineTensor

  3. abstract def padding: Float

  4. abstract def shape: Array[Int]

    Returns the sizes of each dimension of this Tensor.

    Returns the sizes of each dimension of this Tensor.

    Note

    The returned array should be considered as immutable. Changing the array is an undefined behavior.

    ,

    The return value will be Array.empty if this Tensor is a scalar.

Concrete Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. def %(rightHandSide: Tensor): InlineTensor

  4. def *(rightHandSide: Tensor): InlineTensor

  5. def +(rightHandSide: Tensor): InlineTensor

  6. def -(rightHandSide: Tensor): InlineTensor

  7. def ->[B](y: B): (Tensor, B)
    Implicit
    This member is added by an implicit conversion from Tensor to ArrowAssoc[Tensor] performed by method ArrowAssoc in scala.Predef.
    Definition Classes
    ArrowAssoc
    Annotations
    @inline()
  8. def /(rightHandSide: Tensor): InlineTensor

  9. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  10. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  11. def broadcast(newShape: Array[Int]): Tensor

  12. def clone(): AnyRef
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  13. def doCache: Do[CachedTensor]

    Allocates device-side cache that are managed by the RAII.scala library.

  14. def ensuring(cond: (Tensor) ⇒ Boolean, msg: ⇒ Any): Tensor
    Implicit
    This member is added by an implicit conversion from Tensor to Ensuring[Tensor] performed by method Ensuring in scala.Predef.
    Definition Classes
    Ensuring
  15. def ensuring(cond: (Tensor) ⇒ Boolean): Tensor
    Implicit
    This member is added by an implicit conversion from Tensor to Ensuring[Tensor] performed by method Ensuring in scala.Predef.
    Definition Classes
    Ensuring
  16. def ensuring(cond: Boolean, msg: ⇒ Any): Tensor
    Implicit
    This member is added by an implicit conversion from Tensor to Ensuring[Tensor] performed by method Ensuring in scala.Predef.
    Definition Classes
    Ensuring
  17. def ensuring(cond: Boolean): Tensor
    Implicit
    This member is added by an implicit conversion from Tensor to Ensuring[Tensor] performed by method Ensuring in scala.Predef.
    Definition Classes
    Ensuring
  18. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  19. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  20. def finalize(): Unit
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  21. def flatArray: Future[Array[Trees.FloatTrees.FloatTerm.JvmValue]]

    Returns an asynchronous task to read this Tensor into a scala.Array, which is linearized in row-majoy order.

  22. def flatBuffer: Do[FloatBuffer]

    Returns a RAII managed asynchronous task to read this Tensor into an off-heap memory, which is linearized in row-majoy order.

  23. def formatted(fmtstr: String): String
    Implicit
    This member is added by an implicit conversion from Tensor to StringFormat[Tensor] performed by method StringFormat in scala.Predef.
    Definition Classes
    StringFormat
    Annotations
    @inline()
  24. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  25. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  26. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  27. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  28. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  29. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  30. def permute(dimensions: Array[Int]): TransformedTensor

  31. def reshape(newShape: Array[Int]): NonInlineTensor

    Returns a new Tensor of new shape and the same data of this Tensor.

    Returns a new Tensor of new shape and the same data of this Tensor.

    Note

    The data in this Tensor is considered as row-major order when reshape. You can create another column-major version reshape by reversing the shape:

    def columnMajorReshape[Category <: Tensors](tensor: Category#Tensor, newShape: Array[Int]): Category#Tensor = {
  tensor.permute(tensor.shape.indices.reverse.toArray).reshape(newShape.reverse).permute(newShape.indices.reverse.toArray)
}
  32. def scale(newShape: Array[Int]): TransformedTensor

  33. def split(dimension: Int): IndexedSeq[TransformedTensor]

  34. def sum: NonInlineTensor

  35. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  36. def toString(): String

    Definition Classes
    Tensor → AnyRef → Any
  37. def translate(offset: Array[Double], newShape: Array[Int] = shape): TransformedTensor

  38. def unary_+: Tensor.this.type

  39. def unary_-: InlineTensor

  40. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  41. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  42. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @throws( ... )
  43. def →[B](y: B): (Tensor, B)
    Implicit
    This member is added by an implicit conversion from Tensor to ArrowAssoc[Tensor] performed by method ArrowAssoc in scala.Predef.
    Definition Classes
    ArrowAssoc

Shadowed Implicit Value Members

  1. def +(other: String): String
    Implicit
    This member is added by an implicit conversion from Tensor to any2stringadd[Tensor] performed by method any2stringadd in scala.Predef.
    Shadowing
    This implicitly inherited member is shadowed by one or more members in this class.
    To access this member you can use a type ascription:
    (tensor: any2stringadd[Tensor]).+(other)
    Definition Classes
    any2stringadd

Inherited from AnyRef

Inherited from Any

Inherited by implicit conversion any2stringadd from Tensor to any2stringadd[Tensor]

Inherited by implicit conversion StringFormat from Tensor to StringFormat[Tensor]

Inherited by implicit conversion Ensuring from Tensor to Ensuring[Tensor]

Inherited by implicit conversion ArrowAssoc from Tensor to ArrowAssoc[Tensor]

General information

Methods that provides general information of this Tensor.

Slow actions

Actions that can actually perform delayed operations in order to read the data from the device to JVM, or change the internal state of this Tensor.

Delayed operators

Operators that return new Tensors of delay-evaluated computational graphs. The actually computation will be only performed when Slow actions are called.

Ungrouped