将后端从theano更改为tensorflow时如何修改代码?

时间:2019-01-08 04:00:27

标签: tensorflow keras backend theano

这是代码示例: https://github.com/keiserlab/keras-neural-graph-fingerprint/blob/master/NGF/layers.py

开头有

import theano.tensor as T
import keras.backend as K

但是,我想使用tensorflow作为后端。如何修改代码?

此外,后来程序员使用一些类似的代码

# Set pad value and set subtensor of actual tensor
output = T.set_subtensor(output[:, :paddings[0], :], padvalue)
output = T.set_subtensor(output[:, paddings[1]:, :], padvalue)
output = T.set_subtensor(output[:, paddings[0]:x.shape[1] + paddings[0], :], x)

由于我更改了背景,是否还需要更改此代码?

谢谢

0 个答案:

没有答案