public abstract parameter(index as Long) as NDArray<T>
Returns a trainable tensor, ordered like the parameter derivatives.
index
Mutable parameter tensor, not a copy.
weights = layer.parameter(0)