如何在通过loop_vars传递的tf.while_loop体内的变量中赋值？

Question

calrawstate=tf.while_loop(cond,body,[rawstate,x,v,coef])

此处rawstate和v是使用tf.get_variable()创建的两个变量 现在，

def body(rawstate,x,v,coef):
    rawstate.assign_add(tf.matmul(x[v],coef[v]))
    v.assign_add(1)
    return [rawstate,x,v,coef]

如果我尝试做这样的事情，它会引发异常，

AttributeError: 'Tensor' object has no attribute 'assign_add'  

那么如何在tf.while_loop()体内分配值？