Utils

Utilities & Convenience functions…

Convolutions

---

source

conv

 conv (lhs, rhs, window_strides=(1, 1), padding='SAME', **kwargs)

---

source

conv

 conv (lhs, rhs, window_strides=(1, 1), padding='SAME', **kwargs)

---

source

conv

 conv (lhs, rhs, window_strides=(1, 1), padding='SAME', **kwargs)

---

source

dilute

 dilute (touches, brush)

Random

I just can’t be bothered doing this the JAX way…

---

source

randn

 randn (shape, r=None, dtype=<class 'float'>)

---

source

rand

 rand (shape, r=None, dtype=<class 'float'>)

Argmax / Argmin

---

source

argmax2d

 argmax2d (arr2d)

---

source

argmin2d

 argmin2d (arr2d)