tensorflow_backend.py 文件源码

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

项目:DenseNet 作者: titu1994 项目源码 文件源码
def depth_to_space(input, scale, data_format=None):
    ''' Uses phase shift algorithm to convert channels/depth for spatial resolution '''
    if data_format is None:
        data_format = image_data_format()

    if data_format == 'channels_first':
        data_format = 'NCHW'
    else:
        data_format = 'NHWC'

    data_format = data_format.lower()
    out = tf.depth_to_space(input, scale, data_format=data_format)
    return out
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号