basiclib.go

type Dense struct {
	cols   int
	rows   int
	stride int
	data   []float64
}

func NewDense(r, c int, v []float64) *Dense {
	return &Dense{cols: c, rows: r, stride: c, data: v}
}

func (d *Dense) At(i, j int) float64 {
	return d.data[i*d.stride+j]
}

func (d *Dense) Set(i, j int, v float64) {
	d.data[i*d.stride+j] = v
}

func (d *Dense) AtMask(mask [][]int) []float64 {
	out := make([]float64, len(mask))
	for k, v := range mask {
		out[k] = d.At(v[0], v[1])
	}

	return out
}

func (d *Dense) SetMask(mask [][]int, bar []float64) {
	for k, v := range mask {
		d.Set(v[0], v[1], bar[k])
	}
}