Docs / jax4j-core / com.nablatensor.jax4j.nn
final class
Dense
Shared dense layer over the final dimension of any rank, including [B,8,19].
Methods
static Tensor forward(Tensor input, Tensor weight, Tensor bias)
static Var forward(Var input, Var weight, Var bias)
static void into(float[] input, int rows, int in, int out, float[] weight, float[] bias, float[] output, boolean relu)
Allocation-free float32 inference kernel, row-major shared weights. Output must not alias input.