Question

所以我在一个.pb文件中保存了一个tensorflow图，它在我的机器上运行良好，但是当我在tensorflow v 1.3上运行它时，它可以很好地加载，尽管它崩溃并出现以下错误

InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'Reshape' with these attrs.  Registered devices: [CPU], Registered kernels:
  device='CPU'; Tshape in [DT_INT32]
  device='GPU'; T in [DT_COMPLEX128]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_COMPLEX64]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_INT8]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_UINT8]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_INT16]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_UINT16]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_INT64]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_DOUBLE]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_FLOAT]; Tshape in [DT_INT32]
  device='GPU'; T in [DT_HALF]; Tshape in [DT_INT32]

           [[Node: Reshape_165 = Reshape[T=DT_FLOAT, Tshape=DT_INT64](transpose_145, add_31)]]

经过调查，我发现Reshape_165操作的形状为DT_INT64，并且在消息显示时，DT_INT64的v1.3不支持Reshape。想知道是否有一种方法可以将Reshape_165操作dtype更改为DT_INT32，请注意，整个图形都已执行，但是对于此操作，它只是输出concat_52和{{1 }}

concat_53

在Tensorflow图中更改一种节点类型

0 个答案: