import tensorflow as tf
w1 = tf.Variable([[1, 2]]) #int型
res = tf.matmul(w1, [[2],
TensorFlow用tf.gradients()求梯度发生错误
最新推荐文章于 2021-07-05 16:38:18 发布
在使用TensorFlow的tf.gradients()函数时遇到TypeError,原因是输入张量不是float类型而是整型。解决方法是确保所有输入为浮点类型。
摘要由CSDN通过智能技术生成