tensorflow 在`tf.math.unsorted_segment_*`之后，Tensor丢失了形状信息,

nxowjjhe 于 2个月前发布在其他

关注(0)|答案(6)|浏览(47)

请确保这是一个bug。根据我们的规定，我们只处理GitHub上的代码/文档bug、性能问题、功能请求和构建/安装问题。

系统信息：

是否编写了自定义代码(与TensorFlow提供的示例脚本不同):是
OS平台和发行版(例如，Linux Ubuntu 16.04):Linux Ubuntu 16.04
移动设备(例如iPhone 8,Pixel 2,Samsung Galaxy),如果问题发生在移动设备上：否
从哪里安装的TensorFlow(源代码或二进制文件):二进制文件
TensorFlow版本(使用以下命令):2.6.0
Python版本：Python 3.8.5
Bazel版本(如果从源代码编译):否
GCC/编译器版本(如果从源代码编译):否
CUDA/cuDNN版本：11.1/否
GPU型号和内存：GeForce GTX 1070 8117MiB
您可以使用我们的环境捕获工具收集一些此信息。您还可以使用以下命令获取TensorFlow版本：

TF 1.0:python -c "import tensorflow as tf; print(tf.GIT_VERSION, tf.VERSION)"
TF 2.0:python -c "import tensorflow as tf; print(tf.version.GIT_VERSION, tf.version.VERSION)"

当前行为描述：
预期行为描述：

是否希望提交PR?(是/否):否
如果提供候选解决方案，请简要描述：