def build_graph(self, variables):
phs = utils.Utils.map(variables.node, lambda v: tf.placeholder(shape=v.get_shape(), dtype=v.dtype))
self.checked = utils.Utils.map(phs, lambda ph: tf.check_numerics(ph, ''))
return phs
评论列表
文章目录