TensorCreationScope

Properties

Link copied to clipboard
abstract val dtype: KClass<T>
Link copied to clipboard
Link copied to clipboard
abstract val shape: Shape

Functions

Link copied to clipboard
open fun from(vararg data: Float): Tensor<T, V>
open fun from(vararg data: Int): Tensor<T, V>
Link copied to clipboard
open fun fromArray(data: FloatArray): Tensor<T, V>
open fun fromArray(data: IntArray): Tensor<T, V>
Link copied to clipboard
open fun fromIntList(data: List<Int>): Tensor<T, V>
Link copied to clipboard
open fun fromList(data: List<Float>): Tensor<T, V>
Link copied to clipboard
open fun full(value: Number): Tensor<T, V>
Link copied to clipboard
open fun init(generator: (indices: IntArray) -> V): Tensor<T, V>
Link copied to clipboard
open fun ones(): Tensor<T, V>
Link copied to clipboard
open fun randN(mean: Float = 0.0f, std: Float = 1.0f, random: Random = Random.Default): Tensor<T, V>

Create tensor with normal distribution (legacy name)

Link copied to clipboard
open fun randn(mean: Float = 0.0f, std: Float = 1.0f, random: Random = Random.Default): Tensor<T, V>
Link copied to clipboard
open fun random(initBlock: (Shape) -> Tensor<T, V>): Tensor<T, V>

Advanced initialization with custom random distribution.

Link copied to clipboard
open fun randomInit(generator: (random: Random) -> V, random: Random = Random.Default): Tensor<T, V>
Link copied to clipboard
open fun uniform(min: Float = 0.0f, max: Float = 1.0f, random: Random = Random.Default): Tensor<T, V>

Create tensor with uniform distribution

Link copied to clipboard
open fun zeros(): Tensor<T, V>