tensorflow_backend.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:keras-customized 作者: ambrite 项目源码 文件源码
def reverse(x, axes):
    '''Reverse a tensor along the the specified axes
    '''
    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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号