本文整理汇总了Python中tensorflow.python.ops.gen_image_ops._resize_nearest_neighbor_grad方法的典型用法代码示例。如果您正苦于以下问题：Python gen_image_ops._resize_nearest_neighbor_grad方法的具体用法？Python gen_image_ops._resize_nearest_neighbor_grad怎么用？Python gen_image_ops._resize_nearest_neighbor_grad使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.python.ops.gen_image_ops的用法示例。
在下文中一共展示了gen_image_ops._resize_nearest_neighbor_grad方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: _ResizeNearestNeighborGrad
# 需要导入模块: from tensorflow.python.ops import gen_image_ops [as 别名]
# 或者: from tensorflow.python.ops.gen_image_ops import _resize_nearest_neighbor_grad [as 别名]
def _ResizeNearestNeighborGrad(op, grad):
"""The derivatives for nearest neighbor resizing.
Args:
op: The ResizeNearestNeighbor op.
grad: The tensor representing the gradient w.r.t. the output.
Returns:
The gradients w.r.t. the input and the output.
"""
image = op.inputs[0]
if image.get_shape()[1:3].is_fully_defined():
image_shape = image.get_shape()[1:3]
else:
image_shape = array_ops.shape(image)[1:3]
# pylint: disable=protected-access
grads = gen_image_ops._resize_nearest_neighbor_grad(
grad,
image_shape,
align_corners=op.get_attr("align_corners"))
# pylint: enable=protected-access
return [grads, None]