stochman.nnj currently assume that the inputs are vectors (or batches thereof). NNJ should support the standard dimension conventions applied by pytorch.nn, which will open the gates to convolutions and similar operators that are currently missing from stochman.nnj.
stochman.nnj currently assume that the inputs are vectors (or batches thereof). NNJ should support the standard dimension conventions applied by pytorch.nn, which will open the gates to convolutions and similar operators that are currently missing from stochman.nnj.