如何在tensorflow中更改图形操作的数据类型？

Question

我正在尝试构建一个需要Conv2DTranspose的解码器，但是tensorflow iOS没有针对int32类型的操作MUL。有没有办法可以在我的protobuf文件或keras保存的模型中将输入的数据类型更改为MUL？

Answer 1

在创建模型时：

• 使用K.cast(tensor, K.floatx())
• 或tf.cast(tensor, tf.float32)

如果模型已经创建，请复制它，在正确的位置添加操作（在Keras中时，使用Lambda图层：Lambda(lambda x: K.cast(x,K.floatx()))），并传递其权重（{{ 1}}）。