experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix initial test values to avoid numerical instability

Browse Source
This commit is contained in:
Mark Neumann 2017-06-21 12:49:21 -07:00
parent c26146a706
commit 4c75252f25
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/cc/gradients/nn_grad_test.cc
View File

@ -58,10 +58,16 @@ TEST_F(NNGradTest, SoftmaxGrad) {
} }
TEST_F(NNGradTest, LogSoftmaxGrad) { TEST_F(NNGradTest, LogSoftmaxGrad) {
TensorShape shape({32, 10}); TensorShape shape({5, 3});
auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape)); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape));
auto y = LogSoftmax(scope_, x); auto y = LogSoftmax(scope_, x);
RunTest(x, shape, y, shape); // Avoid numerical instability when computing finite differences.
Tensor x_init_value = test::AsTensor<float>(
{-0.9f, -0.7f, -0.5f, -0.3f, -0.1f,
0.1f, 0.3f, 0.5f, 0.7f, 0.8f,
-0.1f, 0.1f, 0.1f, 0.1f, 1.2f},
{5, 3});
RunTest(x, x_init_value, y, shape);
} }
TEST_F(NNGradTest, ReluGrad) { TEST_F(NNGradTest, ReluGrad) {