张量的合并
- concat函数:concat函数用于合并张量,但不会产生新的维度,具体用法为 tf.concat([a,b],axis=x)
其中a和b是两个张量(a,b除了合并的维度之外,其他维度必须相同),x代表在哪个维度上进行合并
- stack函数:concat函数用于合并张量,会产生新的维度,具体用法为 tf.stack([a,b],axis=x)
其中a和b是两个张量(a,b的所有维度必须相同),x代表在哪个维度上增加一个新的维度,并按这个新的维度合并
张量的分割
- unstack函数:unstack函数用于分割张量,会消失一个已有的维度,具体用法为 tf.unstack(a,axis=x) 其中a是张量,x代表在哪个维度上消失一个维度,并按这个消失的维度分割
- split函数:split函数用于分割张量,不会消失一个已有的维度,具体用法为 tf.split(a,axis=x)
其中a是张量,x代表分割后维度的分配情况,数组形式代表具体每个维度分多少,数字形式代表按几均分这个维度