TensorBuilder

class TensorBuilder<T : DType, V>(dtype: KClass<T>)(source)

Builder class for constructing tensors with various initialization strategies

Constructors

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

Functions

Link copied to clipboard
fun <T : DType, V> TensorBuilder<T, V>.bias(): MLPBias<T, V>

Extension function to access MLP bias creation

Link copied to clipboard

Extension function to access image tensor creation

Link copied to clipboard

Extension function to access convolution kernel creation

Link copied to clipboard
fun shape(vararg dimensions: Int): ShapeBuilder<T, V>

Creates a tensor with specified shape and initialization strategy

fun shape(shape: Shape): ShapeBuilder<T, V>

Creates a tensor with specified shape

Link copied to clipboard

Extension function to access MLP weight creation