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

nxowjjhe  于 5个月前  发布在  其他
关注(0)|答案(6)|浏览(66)

请确保这是一个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版本:
  1. TF 1.0:python -c "import tensorflow as tf; print(tf.GIT_VERSION, tf.VERSION)"
  2. TF 2.0:python -c "import tensorflow as tf; print(tf.version.GIT_VERSION, tf.version.VERSION)"

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

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

gcmastyq

gcmastyq1#

请参考类似的错误问题并告知我们:link , link2

nwo49xxi

nwo49xxi2#

请参考类似的错误问题并告知我们:link , link2
我认为与 link2 的问题不同,因为我的代码中根本没有广播操作。

zzzyeukh

zzzyeukh3#

@sonogoto
链接1是否有帮助?

rt4zxlrg

rt4zxlrg4#

@sonogoto 链接1有用吗?
恐怕没有。

oknrviil

oknrviil5#

在Tensorflow 2.7中能够重现您的问题,请查看gist here。谢谢!

jjjwad0x

jjjwad0x6#

Was able to reproduce your issue in Tensorflow 2.7, please find the gist here . Thanks!
It was

相关问题