一、函数定义

通俗易懂些，就是把输入为[batch, height, width, channels]形式的Tensor，其在height和width维的值将移至depth维

space_to_depth(input, block_size, data_format="NHWC", name=None)

input：就是输入的Tensor

block_size：就是块大小，一般≥2

data_format：数据形式，如下表，默认“NHWC”

NHWC	[ batch, height, width, channels ]
NCHW	[ batch, channels, height, width ]
NCHW_VECT_C	qint8 [ batch, channels / 4, height, width, 4 ]

name：名字

*输出张量的深度是block_size * block_size * input_depth

二、解释范例

一共三种Tensor：

（1）输入大小为：[1, 2, 2, 1]

test_tensor_1 = [[[[1], [2]],[[3], [4]]]]

输出大小：[1, 1, 1, 4]

[[[[1 2 3 4]]]]

（2）输入大小为：[1, 2, 2, 3]

test_tensor_2 = [[[[1, 2, 3], [4, 5, 6]],[[7, 8, 9], [10, 11, 12]]]]

输出大小：[1, 1, 1, 12]

[[[[ 1  2  3  4  5  6  7  8  9 10 11 12]]]]

（3）输入大小为：[1, 4, 4, 1]

test_tensor_3 = [[[[1],   [2],  [5],  [6]],[[3],   [4],  [7],  [8]],[[9],  [10], [13],  [14]],[[11], [12], [15],  [16]]]]

输出大小：[1, 2, 2, 4]

[[[[ 1  2  3  4][ 5  6  7  8]][[ 9 10 11 12][13 14 15 16]]]]

三、代码验证

test_space_to_depth.py

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'test_tensor_1 = tf.placeholder('float32', [1, 2, 2, 1], name='test_tensor_1')
test_tensor_1 = [[[[1], [2]],[[3], [4]]]]
test_tensor_1 = tf.space_to_depth(test_tensor_1, 2)test_tensor_2 = tf.placeholder('float32', [1, 2, 2, 3], name='test_tensor_1')
test_tensor_2 = [[[[1, 2, 3], [4, 5, 6]],[[7, 8, 9], [10, 11, 12]]]]
test_tensor_2 = tf.space_to_depth(test_tensor_2, 2)test_tensor_3 = tf.placeholder('float32', [1, 4, 4, 1], name='test_tensor_1')
test_tensor_3 = [[[[1],   [2],  [5],  [6]],[[3],   [4],  [7],  [8]],[[9],  [10], [13],  [14]],[[11], [12], [15],  [16]]]]
test_tensor_3 = tf.space_to_depth(test_tensor_3, 2)sess = tf.Session()
print(test_tensor_1)
print(sess.run(test_tensor_1))print(test_tensor_2)
print(sess.run(test_tensor_2))print(test_tensor_3)
print(sess.run(test_tensor_3))

结果：

TensorFlow中 tf.space_to_depth()函数的用法相关推荐

tensorflow中tf.get_variable()函数详解
如果变量存在,函数tf.get_variable()会返回现有的变量:如果变量不存在,会根据给定形状和初始值创建一个新的变量. def get_variable(name, shape=None, d ...
TF版本升级问题集合：成功解决由于不同的TensorFlow版本之间某些函数的用法引起的错误
TF版本升级问题集合:成功解决由于不同的TensorFlow版本之间某些函数的用法引起的错误目录遇到的问题以及对应解决方法遇到的问题以及对应解决方法 Transitioning to Tenso ...
【TensorFlow】tf.nn.softmax_cross_entropy_with_logits的用法
[TensorFlow]tf.nn.softmax_cross_entropy_with_logits的用法 from:https://blog.csdn.net/mao_xiao_feng/arti ...
【Tensorflow】Tensorflow中的卷积函数（conv2d、slim.conv2d、depthwise_conv2d、conv2d_transpose）
[fishing-pan:https://blog.csdn.net/u013921430 转载请注明出处] 前言卷积是卷积神经网络中最主要.最重要的运算.想必大家最早接触卷积的概念就是在初高中 ...
tensorflow中的norm()函数 | axis=0,axis=1,axis=2
tensorflow中的norm函数作用是用来求L1_norm范数和Eukl_norm范数. 本篇文章主要目的是讲解axis的含义,所以默认都是用L1_norm范数,方便理解. 首先,来看一下2维的情 ...
MapInfo中常用查询函数及用法
MapInfo中常用查询函数及用法: 函数用途语法备注图层中选点 Str$(obj)="point": Str(String)表示字符串:point表示点: 图层中选线 St ...
python join_详解Python中的join()函数的用法
函数:string.join() Python中有join()和os.path.join()两个函数,具体作用如下: join(): 连接字符串数组.将字符串.元组.列表中的元素以指定的字符(分隔符) ...
php中msubstr,PHP学习：thinkphp中字符截取函数msubstr()用法分析
<PHP学习:thinkphp中字符截取函数msubstr()用法分析>要点: 本文介绍了PHP学习:thinkphp中字符截取函数msubstr()用法分析,希望对您有用.如果有疑问,可 ...
matlab fopen函数的用法,matlab中关于fread函数的用法
公告: 为响应国家净网行动,部分内容已经删除,感谢读者理解. 话题:matlab中关于fread函数的用法问题详情:用fread打开一个时,需要不需要把开放在某个特回答:必须在Matlab可以搜索的 ...

TensorFlow中 tf.space_to_depth()函数的用法

一、函数定义

二、解释范例

三、代码验证

TensorFlow中 tf.space_to_depth()函数的用法相关推荐

最新文章

热门文章