本文整理汇总了Python中tensorflow.contrib.staging.StagingArea方法的典型用法代码示例。如果您正苦于以下问题:Python staging.StagingArea方法的具体用法?Python staging.StagingArea怎么用?Python staging.StagingArea使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.contrib.staging
的用法示例。
在下文中一共展示了staging.StagingArea方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: _create_staging_area
# 需要导入模块: from tensorflow.contrib import staging [as 别名]
# 或者: from tensorflow.contrib.staging import StagingArea [as 别名]
def _create_staging_area(self, tensors):
# return tf_staging.StagingArea(dtypes=[tensor.dtype for tensor in tensors], shapes=[tensor.shape for tensor
# in tensors])
names, dtypes, shapes = [], [], []
for name, tensor in tensors.items():
dtypes.append(tensor.dtype)
shapes.append(tensor.shape)
names.append(name)
return tf_staging.StagingArea(dtypes=dtypes, shapes=shapes, names=names)
示例2: _create_staging_area
# 需要导入模块: from tensorflow.contrib import staging [as 别名]
# 或者: from tensorflow.contrib.staging import StagingArea [as 别名]
def _create_staging_area(self, tensors):
return tf_staging.StagingArea(dtypes=[tensor.dtype for tensor in tensors], shapes=[tensor.shape for tensor in tensors])
示例3: _create_staging_area
# 需要导入模块: from tensorflow.contrib import staging [as 别名]
# 或者: from tensorflow.contrib.staging import StagingArea [as 别名]
def _create_staging_area(self, tensors):
names, dtypes, shapes = [], [], []
for name, tensor in tensors.items():
dtypes.append(tensor.dtype)
shapes.append(tensor.shape)
names.append(name)
return tf_staging.StagingArea(dtypes=dtypes, shapes=shapes, names=names)