skainet-lang-dag/sk.ainet.lang.dag/DagBuilder

DagBuilder

class DagBuilder(source)

Builder that records symbolic nodes/values. It is definition-only: no tensors are allocated.

Constructors

DagBuilder

constructor()

Functions

abs

fun DagBuilder.abs(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for abs.

add

fun DagBuilder.add(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for add.

addScalar

fun DagBuilder.addScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for addScalar.

avgPool2d

fun DagBuilder.avgPool2d(input: GraphValue<*>, kernelSize: Pair<Int, Int>, stride: Pair<Int, Int> = 1 to 1, padding: Pair<Int, Int> = 0 to 0, countIncludePad: Boolean, id: String = ""): GraphValue<*>

DSL extension for avgPool2d.

clamp

fun DagBuilder.clamp(tensor: GraphValue<*>, minVal: Float, maxVal: Float, id: String = ""): GraphValue<*>

DSL extension for clamp.

concat

fun DagBuilder.concat(tensors: List<GraphValue<*>>, dim: Int, id: String = ""): GraphValue<*>

DSL extension for concat.

constant

inline fun <T : DType, V> constant(name: String, noinline builder: SymbolicTensorBuilder<T>.() -> TensorSpec): GraphValue<T>

Constant helper that reuses a symbolic, allocation-free data DSL to declare shape/dtype.

fun <T : DType> constant(name: String, spec: TensorSpec): GraphValue<T>

Declare a constant placeholder (treated like an input node).

conv1d

fun DagBuilder.conv1d(input: GraphValue<*>, weight: GraphValue<*>, bias: GraphValue<*>, stride: Int, padding: Int, dilation: Int, groups: Int = 1, id: String = ""): GraphValue<*>

DSL extension for conv1d.

conv2d

fun DagBuilder.conv2d(input: GraphValue<*>, weight: GraphValue<*>, bias: GraphValue<*>, stride: Pair<Int, Int> = 1 to 1, padding: Pair<Int, Int> = 0 to 0, dilation: Pair<Int, Int> = 1 to 1, groups: Int = 1, id: String = ""): GraphValue<*>

DSL extension for conv2d.

conv3d

fun DagBuilder.conv3d(input: GraphValue<*>, weight: GraphValue<*>, bias: GraphValue<*>, stride: Triple<Int, Int, Int>, padding: Triple<Int, Int, Int>, dilation: Triple<Int, Int, Int>, groups: Int = 1, id: String = ""): GraphValue<*>

DSL extension for conv3d.

divide

fun DagBuilder.divide(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for divide.

divScalar

fun DagBuilder.divScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for divScalar.

elu

fun DagBuilder.elu(tensor: GraphValue<*>, alpha: Float, id: String = ""): GraphValue<*>

DSL extension for elu.

flatten

fun DagBuilder.flatten(tensor: GraphValue<*>, startDim: Int = 0, endDim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for flatten.

fun DagBuilder.ge(tensor: GraphValue<*>, value: Float, id: String = ""): GraphValue<*>

DSL extension for ge.

gelu

fun DagBuilder.gelu(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for gelu.

input

fun <T : DType> input(name: String, spec: TensorSpec = TensorSpec(name = name, shape = null, dtype = "unknown")): GraphValue<T>

Declare a graph input placeholder.

leakyRelu

fun DagBuilder.leakyRelu(tensor: GraphValue<*>, negativeSlope: Float, id: String = ""): GraphValue<*>

DSL extension for leakyRelu.

logSoftmax

fun DagBuilder.logSoftmax(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for logSoftmax.

fun DagBuilder.lt(tensor: GraphValue<*>, value: Float, id: String = ""): GraphValue<*>

DSL extension for lt.

matmul

fun DagBuilder.matmul(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for matmul.

maxPool2d

fun DagBuilder.maxPool2d(input: GraphValue<*>, kernelSize: Pair<Int, Int>, stride: Pair<Int, Int> = 1 to 1, padding: Pair<Int, Int> = 0 to 0, id: String = ""): GraphValue<*>

DSL extension for maxPool2d.

mean

fun DagBuilder.mean(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for mean.

module

fun DagBuilder.module(module: DagModule, inputs: List<GraphValue<*>>): List<GraphValue<*>>

DSL helper for using modules.

mulScalar

fun DagBuilder.mulScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for mulScalar.

multiply

fun DagBuilder.multiply(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for multiply.

narrow

fun DagBuilder.narrow(tensor: GraphValue<*>, dim: Int, start: Int, length: Int, id: String = ""): GraphValue<*>

DSL extension for narrow.

fun op(operation: Operation, inputs: List<GraphValue<*>>, id: String = "", attributes: Map<String, Any?> = emptyMap()): List<GraphValue<*>>

Generic operation hook that lets callers wire custom Operation instances.

output

fun output(vararg values: GraphValue<*>)

Mark a value as a program output. If none are marked, the last node's outputs are used.

pad2d

fun DagBuilder.pad2d(tensor: GraphValue<*>, padLeft: Int, padRight: Int, padTop: Int, padBottom: Int, id: String = ""): GraphValue<*>

DSL extension for pad2d.

parameter

inline fun <T : DType, V> parameter(name: String, noinline builder: SymbolicTensorBuilder<T>.() -> TensorSpec): GraphValue<T>

Parameter helper that reuses a symbolic, allocation-free data DSL to declare shape/dtype.

fun <T : DType> parameter(name: String, spec: TensorSpec): GraphValue<T>

Declare a parameter/weight placeholder.

rdivScalar

fun DagBuilder.rdivScalar(a: Number, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for rdivScalar.

relu

fun DagBuilder.relu(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for relu.

reshape

fun DagBuilder.reshape(tensor: GraphValue<*>, newShape: Shape, id: String = ""): GraphValue<*>

DSL extension for reshape.

rsubScalar

fun DagBuilder.rsubScalar(a: Number, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for rsubScalar.

sigmoid

fun DagBuilder.sigmoid(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for sigmoid.

sign

fun DagBuilder.sign(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for sign.

silu

fun DagBuilder.silu(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for silu.

softmax

fun DagBuilder.softmax(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for softmax.

sqrt

fun DagBuilder.sqrt(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for sqrt.

squeeze

fun DagBuilder.squeeze(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for squeeze.

subScalar

fun DagBuilder.subScalar(a: GraphValue<*>, b: Number, id: String = ""): GraphValue<*>

DSL extension for subScalar.

subtract

fun DagBuilder.subtract(a: GraphValue<*>, b: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for subtract.

sum

fun DagBuilder.sum(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for sum.

transpose

fun DagBuilder.transpose(tensor: GraphValue<*>, id: String = ""): GraphValue<*>

DSL extension for transpose.

tril

fun DagBuilder.tril(tensor: GraphValue<*>, k: Int, id: String = ""): GraphValue<*>

DSL extension for tril.

unfold

fun DagBuilder.unfold(tensor: GraphValue<*>, dim: Int, size: Int, step: Int, id: String = ""): GraphValue<*>

DSL extension for unfold.

unsqueeze

fun DagBuilder.unsqueeze(tensor: GraphValue<*>, dim: Int, id: String = ""): GraphValue<*>

DSL extension for unsqueeze.

upsample2d

fun DagBuilder.upsample2d(input: GraphValue<*>, scale: Pair<Int, Int>, mode: UpsampleMode = UpsampleMode.Nearest, alignCorners: Boolean = false, id: String = ""): GraphValue<*>

DSL extension for upsample2d.

variance

fun DagBuilder.variance(tensor: GraphValue<*>, dim: Int = -1, id: String = ""): GraphValue<*>

DSL extension for variance.