Reduce rank and set values according to 3rd dimension

SameDiff
10

@SidneyLann is this for weighted loss? If not why not make that a placeholder?

"Attention Is All You Need" model implementation using dl4j