Welvet examples
20. CNN family — cnn1 / cnn2 / cnn3
When: spatial / temporal convolution senses (1D seq, 2D images, 3D volumes).
Where: layers/cnn1, layers/cnn2, layers/cnn3
Why: im2col → Dense Proj so quant/SIMD/WebGPU and CamSync (via Proj) are shared.
cd 20-cnn && source ../env.sh && go run .
| Op | Use |
|---|---|
| cnn1 | Audio / 1D sequences |
| cnn2 | Images / MNIST-style |
| cnn3 | Volumetric / video voxels |
Put CNN inside Parallel cams when you want CamSync on kernels.
main.go
Download source ↓package main
import (
"fmt"
"github.com/openfluke/welvet/core"
"github.com/openfluke/welvet/layers/cnn1"
"github.com/openfluke/welvet/layers/cnn2"
"github.com/openfluke/welvet/layers/cnn3"
)
func main() {
// cnn1 — 1D conv (seq / audio patches)
c1, err := cnn1.New(cnn1.Config{InChannels: 1, Filters: 4, SeqLen: 16, Kernel: 3, Stride: 1})
must(err)
x1 := core.NewTensor[float32](1, 1, 16)
for i := range x1.Data {
x1.Data[i] = 0.1
}
_, y1, err := cnn1.Forward(c1, x1)
must(err)
fmt.Println("cnn1", y1.Shape)
// cnn2 — 2D vision
c2, err := cnn2.New(cnn2.Config{InChannels: 1, Filters: 4, Height: 8, Width: 8, Kernel: 3, Stride: 1})
must(err)
x2 := core.NewTensor[float32](1, 1, 8, 8)
for i := range x2.Data {
x2.Data[i] = 0.1
}
_, y2, err := cnn2.Forward(c2, x2)
must(err)
fmt.Println("cnn2", y2.Shape)
// cnn3 — 3D / volumetric
c3, err := cnn3.New(cnn3.Config{InChannels: 1, Filters: 2, Depth: 4, Height: 4, Width: 4, Kernel: 2, Stride: 1})
must(err)
x3 := core.NewTensor[float32](1, 1, 4, 4, 4)
for i := range x3.Data {
x3.Data[i] = 0.1
}
_, y3, err := cnn3.Forward(c3, x3)
must(err)
fmt.Println("cnn3", y3.Shape)
}
func must(err error) {
if err != nil {
panic(err)
}
}