小编Kae*_*ede的帖子

创建一个张量原型,其内容大于2GB

我创建了一个大小为(2 ^ 22,256)的ndarray(W),并尝试使用以下数组作为权重矩阵的初始化:

w = tf.Variable(tf.convert_to_tensor(W))
Run Code Online (Sandbox Code Playgroud)

然后,tensorflow引发了一个错误: ValueError:无法创建内容大于2GB的张量原型。

我该如何解决这个问题?PS。我的权重矩阵必须使用该(2 ^ 22,256)矩阵进行初始化。谢谢 :)

python tensorflow

3
推荐指数
1
解决办法
3719
查看次数

标签 统计

python ×1

tensorflow ×1