INDArray function ( INDArray arg0 , INDArray arg1 ) { INDArray loc0 = arg1 . mul ( arg0 ) . sum ( 1 ) ; return arg1 . mul ( arg0 . subColumnVector ( loc0 ) ) ; }
calculate dl/dz for softmax activation function , from dl/da and a , where a : output activations dl/da : derivative of loss function with respect to the output activations note : this version does not modify either input arrays . this is less efficient . for a more efficient implementation that modifies the input arrays , use #dldzsoftmaxi indarray , indarray . concode_field_sep PlaceHolder placeHolder concode_field_sep INDArray dLdZsoftmaxi concode_elem_sep INDArray dLdZsoftmaxPreOut