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
tensorflow_backend.py 文件源码
python
阅读 25
收藏 0
点赞 0
评论 0
评论列表
文章目录