以下是我的代码
import tensorflow as tf
import numpy as np
def forward(x):
z = tf.Variable(tf.zeros_like(x), trainable=False)
s = tf.shape(x)[0]
for i in range(s):
z[i].assign(x[i]**i)
return z
a = tf.Variable(np.ones([5])*3)
with tf.GradientTape() as tape:
b = forward(a)
grad = tape.gradient(b, a)
我有一个输入,我必须对其进行切片然后计算输出。在此基础上,我需要计算梯度。但是,上述代码的输出为 None。
如何获得渐变?有什么方法可以对输入进行切片以获得渐变。
PS我必须只使用EagerExecution。没有图表模式。