TensorFactoryContext

class TensorFactoryContext<T : DType, V>(executionContext: ExecutionContext, dtype: KClass<T>)(source)

Context wrapper that provides tensor data factory access

Constructors

Link copied to clipboard
constructor(executionContext: ExecutionContext, dtype: KClass<T>)

Functions

Link copied to clipboard
fun shape(vararg dimensions: Int, init: TensorCreationScope<T, V>.(Shape) -> Tensor<T, V>): Tensor<T, V>
fun shape(shape: Shape, init: TensorCreationScope<T, V>.(Shape) -> Tensor<T, V>): Tensor<T, V>

Create tensor with specified shape and initialization strategy