com.thoughtworks.compute.Tensors
FillTensor
Related Doc:
package Tensors
Allocates device-side cache that are managed by the RAII.scala library.
Allocates device-side cache that are managed by the RAII.scala library.
Returns an asynchronous task to read this Tensor into a scala.Array, which is linearized in row-major order.
Returns an asynchronous task to read this Tensor into a scala.Array, which is linearized in row-major order.
Returns a RAII managed asynchronous task to read this Tensor into an off-heap memory, which is linearized in row-major order.
Returns an asynchronous task to read this Tensor into a scala.Array
Returns an asynchronous task to read this Tensor into a scala.Array
Returns an asynchronous task to read this Tensor into a scala.Seq
Returns an asynchronous task to read this Tensor into a 2D scala.Array
Returns an asynchronous task to read this Tensor into a 2D scala.Array
Returns an asynchronous task to read this Tensor into a 2D scala.Seq
Returns an asynchronous task to read this Tensor into a 3D scala.Array
Returns an asynchronous task to read this Tensor into a 3D scala.Array
Returns an asynchronous task to read this Tensor into a 3D scala.Seq
Returns an asynchronous task to read this Tensor into a 4D scala.Array
Returns an asynchronous task to read this Tensor into a 4D scala.Array
Returns an asynchronous task to read this Tensor into a 4D scala.Seq
Returns an asynchronous task to read this Tensor into a 5D scala.Array
Returns an asynchronous task to read this Tensor into a 5D scala.Array
Returns an asynchronous task to read this Tensor into a 5D scala.Seq
Returns an asynchronous task to read this Tensor into a scala.Float
Returns an asynchronous task to read this Tensor into a scala.Float
Returns a new Tensor of new shape and the same data of this Tensor.
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) }
(fillTensor: any2stringadd[FillTensor]).+(other)
Methods that provides general information of this Tensor.
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.
Operators that return new Tensors of delay-evaluated computational graphs. The actually computation will be only performed when Slow actions are called.