目录
4.tf.reshape(tensor,shape,name) 转换维度
5. tf.pad(tensor,paddings,mode,name,constant_values) 张量值填充
1.tf.split 切割张量
API原型(TensorFlow 1.8.0):
tf.split(
value,
num_or_size_splits,
axis=0,
num=None,
name='split'
)
这个函数是用来切割张量的。输入切割的张量和参数,返回切割的结果。 value
传入的就是需要切割的张量。
这个函数有两种切割的方式:比如说一个20 * 30 * 40的张量
有两种切割方式:
1. 如果num_or_size_splits
传入的是一个整数,这个整数代表这个张量最后会被切成几个小张量。此时,传入axis
的数值就代表切割哪个维度(从0开始计数)。调用tf.split(my_tensor, 2,0)
返回两个10 * 30 * 40的小张量。
2. 如果num_or_size_splits
传入的是一个向量,那么向量有几个分量就分成几份,切割的维度还是由axis
决定。比如调用tf.split(my_tensor, [10, 5, 25], 2)
,则返回三个张量分别大小为 20 * 30 * 10、20 * 30 * 5、20 * 30 * 25。很显然,传入的这个向量各个分量加和必须等于axis
所指示原张量维度的大