Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
ND4J Activation functions: enforce same shape in backprop method #5357
Yet: we don't actually enforce this properly at the ActivationFunction level. Some of the internal ops may throw an exception, however (but this is not guaranteed for all cases it seems).