Fix softmax unittest (#25371)

* test=develop, fix softmax
fix_copy_if_different
littletomatodonkey 5 years ago committed by GitHub
parent 2989c012f2
commit 64b4612290
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -47,6 +47,7 @@ class TestSoftmaxOp(OpTest):
self.shape = self.get_x_shape()
self.axis = self.get_axis()
np.random.seed(0)
x = np.random.uniform(0.1, 1, self.shape).astype(self.dtype)
out = np.apply_along_axis(stable_softmax, self.axis, x)

Loading…
Cancel
Save