TensorHelpers

lamp.TensorHelpers
object TensorHelpers

Attributes

Graph
Supertypes
class Object
trait Matchable
class Any
Self type

Members list

Value members

Concrete methods

def device(t: Tensor): Device
def fromByteArray(arr: Array[Byte], dim: Seq[Long], device: Device): Tensor
def fromDoubleArray(arr: Array[Double], dim: Seq[Long], device: Device, precision: FloatingPointPrecision): Tensor
def fromFloatArray(arr: Array[Float], dim: Seq[Long], device: Device): Tensor
def fromIntArray(arr: Array[Int], dim: Seq[Long], device: Device): Tensor
def fromLongArray(arr: Array[Long], device: Device): Tensor
def fromLongArray(arr: Array[Long], dim: Seq[Long], device: Device): Tensor
def fromLongArrayOfArrays(arr: Array[Array[Long]], dim: Seq[Long], device: Device): Tensor
def fromShortArray(arr: Array[Short], dim: Seq[Long], device: Device): Tensor
def precision(t: Tensor): Option[SinglePrecision.type | DoublePrecision.type]
def toByteArray(t0: Tensor): Array[Byte]
def toDoubleArray(t0: Tensor): Array[Double]
def toFloatArray(t0: Tensor): Array[Float]
def toIntArray(t0: Tensor): Array[Int]
def toLongArray(t0: Tensor): Array[Long]
def toShortArray(t0: Tensor): Array[Short]
def unbroadcast(p: Tensor, desiredShape: List[Long]): Option[Tensor]