正如我所看到的,tensorflow中bfloat16有一个numpy扩展:
可以通过调用RegisterNumpyBfloat16
来启用此扩展。或者至少应该这样。我已安装tensorflow 2.4.1并尝试启用该扩展,但出现以下错误:
> tf.RegisterNumpyBfloat16()
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-6-4f89beb80796> in <module>
----> 1 tf.RegisterNumpyBfloat16()
AttributeError: module 'tensorflow' has no attribute 'RegisterNumpyBfloat16'
有人知道我做错了什么吗?或者如何启用此numpy扩展
多谢各位
我刚刚在TensorFlow 2.4.0和NumPy 1.19.4上试过这个
试试这个:
相关问题 更多 >
编程相关推荐