tf.split(
value,
num_or_size_splits,
axis=0,
num=None,
name='split'
)
把张量分解成子张量。如果num_or_size_split是一个整数,那么值将沿着维度轴拆分为num_split较小的张量。这要求num_split均匀地分割value.shape[axis]。如果num_or_size_split是一个一维张量(或列表),称之为size_split,值被分割成len(size_split)元素。第i个元素的形状与值的大小相同,除了尺寸为size_split [i]的维度轴。
例如:
# 'value' is a tensor with shape [5, 30]
# Split 'value' into 3 tensors with sizes [4, 15, 11] along dimension 1
split0, split1, split2 = tf.split(value, [4, 15, 11], 1)
tf.shape(split0) # [5, 4]
tf.shape(split1) # [5, 15]
tf.shape(split2) # [5, 11]
# Split 'value' into 3 tensors along dimension 1
split0, split1, split2 = tf.split(value, num_or_size_splits=3, axis=1)
tf.shape(split0) # [5, 10]
参数:
value
: 要分割的张量。返回值:
异常:
ValueError
: If num
is unspecified and cannot be inferred.原链接: https://tensorflow.google.cn/versions/r1.14/api_docs/python/tf/split