tf.concat用于将多个张量在某维度合并起来,类似于numpy.concatenate。第一个参数为待合并的多个张量列表,第二个参数需指定哪个维度上做合并。待合并的张量shape需要完全相同,并且产生的张量的阶数不会发生变化。

tensorflow随笔——concat(), stack(), unstack()

tf.stack(values, axis=0, name='stack')以指定的轴axis,将一个维度为R的张量数组转变成一个维度为R+1的张量。即会在新的张量阶上合并,张量的阶数也会增加。

参数values:一个具有相同形状与数据类型的张量数组。

参数axis:stack就是要将一组相同形状的张量提高一个维度,axis就是这些张量里,将axis指定的维度用所有这些张量数组代替。如axis=2,表示指定在第2个维度,原来的元素用整个张量数组里的元素代替,即(A,B,C)转变成(A,B,N,C)。

参数name:操作的名字。

tensorflow随笔——concat(), stack(), unstack()

tf.unstack(value, num=None, axis=0, name='unstack')以指定的轴axis,将一个维度为R的张量数组转变成一个维度为R-1的张量。即将一组张量以指定的轴,减少一个维度。

参数value:一个将要被降维的维度大于0的张量

参数num:指定的维度axis的长度。如果设置为None将自动取值

参数axis:unstack就是要将一个张量降低为低一个维度的张量。axis就是将指定的维度用所有这个张量里同维度的数据代替。

参数name:操作的名字

tensorflow随笔——concat(), stack(), unstack()

 

 

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2021-07-31
  • 2022-12-23
  • 2022-12-23
  • 2021-07-23
  • 2021-12-12
猜你喜欢
  • 2021-07-06
  • 2021-07-10
  • 2021-08-22
  • 2022-12-23
  • 2021-04-25
相关资源
相似解决方案