《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。