ShapeBuilder

Builder for tensors with a defined shape, allowing initialization

Constructors

constructor(dtype: KClass<T>, shape: Shape)

Initialize tensor with a constant value

Extension function to access image-specific initializers

fun init(generator: (indices: IntArray) -> V): TensorInitializer<T, V>

Initialize tensor with custom function

Initialize tensor with ones

fun randn(mean: Float = 0.0f, std: Float = 1.0f, random: Random = Random.Default): TensorInitializer<T, V>

Initialize tensor with normal distribution

fun randomInit(generator: (random: Random) -> V, random: Random = Random.Default): TensorInitializer<T, V>

Initialize tensor with custom random distribution

fun uniform(min: Float = 0.0f, max: Float = 1.0f, random: Random = Random.Default): TensorInitializer<T, V>

Initialize tensor with uniform distribution

Initialize tensor with zeros