def assertSparseOutput(self, expected_indices, expected_values,
expected_shape, actual_sparse_tensor, close_values):
with tf.Session() as sess:
sess.run(tf.tables_initializer())
actual = actual_sparse_tensor.eval()
self.assertAllEqual(expected_indices, actual.indices)
self.assertAllEqual(expected_shape, actual.dense_shape)
if close_values:
self.assertAllClose(expected_values, actual.values)
else:
self.assertAllEqual(expected_values, actual.values)
评论列表
文章目录