Блог

Главная - Вопросы по программированию - Измените тип одного узла в графике Tensorflow

Измените тип одного узла в графике Tensorflow

#tensorflow #tensor

#тензорный поток #тензор

Вопрос:

Итак, у меня есть график tensorflow, сохраненный в .pb файле, он хорошо работает на моей машине, но когда я запускаю его на 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 , а версия v1.3 не поддерживает DT_INT64 для Reshape , как указано в сообщении, поэтому мне было интересно, есть ли способ изменить Reshape_165 тип операции dtype, чтобы DT_INT32 иметь в виду, что выполняется весь график, но для этой операции, которая выполняется непосредственно перед выводами concat_52 и concat_53 

 add_31
Reshape_165
concat_52/axis
concat_52
concat_53/axis
concat_53