在TensorFlow中调试python测试

Question

我们希望在TensorFlow (如sparse_split_op_test和string_to_hash_bucket_op_test )中调试Python测试

其他c++测试可以使用gdb进行调试，但是我们无法找到调试python测试的方法。

我们是否可以通过Bazel test命令(例如bazel test //tensorflow/python/kernel_tests:sparse_split_op_test) )来调试特定的python测试用例？

Answer 1

我首先要构建一个测试：

bazel build //tensorflow/python/kernel_tests:sparse_split_op_test

然后对生成的Python二进制文件使用pdb：

pdb bazel-bin/tensorflow/python/kernel_tests/sparse_split_op_test

这似乎适用于我通过测试的前几行。