minor javadoc change
The weights are now returning shapedArray, updating also functions that use weights