《Born》附录A：API 速查表

本附录提供 Born 核心包的常用 API 快速查阅。

tensor 包

创建张量

Zeros[T, B](shape, backend)
Ones[T, B](shape, backend)
Randn[T, B](shape, backend)
FromSlice[T, B](data, shape, backend)
Arange[T, B](start, end, backend)

逐元素运算

Add, Sub, Mul, Div
Exp, Log, Sqrt, ReLU, Sigmoid, Tanh
Softmax(dim), Clamp(min, max)

矩阵与形状

MatMul, Transpose, Reshape
Squeeze, Unsqueeze, Cat, Chunk
Conv2D, MaxPool2D, Gather

规约

Sum, SumDim(dim), MeanDim(dim), Argmax(dim)

nn 包

模块

NewLinear(in, out, backend)
NewConv2D(in, out, kernel, backend)
NewReLU(), NewSiLU(), NewSigmoid()
NewLayerNorm(size, backend)
NewMHA(dModel, nHead, backend)
NewGQA(dModel, nHead, nKVHead, backend)
NewTransformerBlock(config, backend)

损失函数

NewCrossEntropyLoss(backend)
NewMSELoss(backend)

工具

nn.Save(model, path, name, metadata)
nn.Load(path, backend, model)
nn.Accuracy(output, target, topK)

optim 包

NewSGD(lr, params...)
NewAdam(lr, params...)
NewAdamW(lr, weightDecay, params...)

optimizer.Step()
optimizer.ZeroGrad()
optimizer.SetLR(lr)

环境变量

📘 《Born》连载技术书，附录 A/4。