切片张量是否在Tensorflow中提供渐变?

我想将1024x1024图像张量切成16个大小为256x256的张量。我正在使用以下代码进行操作:

with tf.variable_scope("image_split",reuse=tf.AUTO_REUSE):
    img_array = tf.get_variable(name = 'img_array',shape = (16,256,3),dtype = tf.float32)     
with tf.control_dependencies([img_array[i].assign(img[(i%4)*256:(i%4+1)*256,(i//4)*256:((i+4)//4)*256]) for i in range(16)]):
        img_array = tf.identity(img_array)

但是我收到此错误:ValueError: No gradients provided for any variable,check your graph for ops that do not support gradients

如果我不切片图像,则一切正常。切片张量是否提供渐变?是的话,正确的方法是什么?

wodeqiji110 回答:切片张量是否在Tensorflow中提供渐变?

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3156821.html

大家都在问