tensorflow:注册numpy bfloat16扩展
tensorflow:注册numpy bfloat16扩展
提问于 2021-02-24 23:35:36
正如我所看到的，在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扩展？
非常感谢
回答 2
Stack Overflow用户
回答已采纳
发布于 2021-02-25 02:07:51
试试这个：
from tensorflow.python import _pywrap_bfloat16
bfloat16 = _pywrap_bfloat16.TF_bfloat16_type()
print(bfloat16)
# <class 'bfloat16'>
print(bfloat16(1.0))
# bfloat16(1)Stack Overflow用户
发布于 2021-03-23 11:46:06
我刚刚在TensorFlow 2.4.0和NumPy 1.19.4上尝试了一下。
import numpy as np
import tensorflow as tf
bfloat16 = tf.bfloat16.as_numpy_dtype
np.array([1.0, 2.0, 3.0], dtype=bfloat16)
# array([bfloat16(1), bfloat16(2), bfloat16(3)], dtype=bfloat16)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/66354058
复制相关文章
相似问题
腾讯云开发者
