他的回复:
在2.x版本上使用tf.Variable创建变量,同时在var和accum上加上.handle方法import numpy as npimport tensorflow as tfvar = tf.Variable(np.random.rand(2, 1).astype(np.float32))accum = tf.Variable(np.random.rand(2, 1).astype(np.float32))lr = 0.01grad = tf.Variable(np.random.rand(2, 1).astype(np.float32))momentum=0.0res = tf.raw_ops.ResourceApplyKerasMomentum(var=var.handle, accum=accum.handle, lr=lr, grad=grad, momentum=momentum, use_locking=False, use_nesterov=False) print("==tf res==", res)