def reverse(x, axes):
"""Reverse a tensor along the the specified axes
# Returns
A tensor.
"""
if isinstance(axes, int):
axes = [axes]
try:
return tf.reverse_v2(x, axes)
except AttributeError:
# Older TF versions.
dims = [True if i in axes else False for i in range(len(x.get_shape()._dims))]
return tf.reverse(x, dims)
# VALUE MANIPULATION
评论列表
文章目录