lamp-sten/lamp/TensorHelpers

TensorHelpers

object TensorHelpers
class Object
trait Matchable
class Any

Value members

Concrete methods

def device(t: Tensor): Device
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 fromLongArray(arr: Array[Long], device: Device): Tensor
def fromLongArray(arr: Array[Long], dim: Seq[Long], device: Device): Tensor
def precision(t: Tensor): Option[SinglePrecision.type | DoublePrecision.type]
def toDoubleArray(t0: Tensor): Array[Double]
def toFloatArray(t0: Tensor): Array[Float]
def toLongArray(t0: Tensor): Array[Long]
def unbroadcast(p: Tensor, desiredShape: List[Long]): Option[Tensor]