合并

tflearn.layers.merge_ops.merge (tensors_list, mode, axis=1, name='Merge')

Tensor 列表合并成一个。必须指定合并“模式”,请查看以下不同选项。

输入

张量列表。

输出

合并后的张量。

参数

  • tensors_list: Tensor 列表,要合并的张量列表。
  • mode: str。合并模式,支持以下几种:
'concat': concatenate outputs along specified axis
'elemwise_sum': outputs element-wise sum
'elemwise_mul': outputs element-wise sum
'sum': outputs element-wise sum along specified axis
'mean': outputs element-wise average along specified axis
'prod': outputs element-wise multiplication along specified axis
'max': outputs max elements along specified axis
'min': outputs min elements along specified axis
'and': `logical and` btw outputs elements along specified axis
'or': `logical or` btw outputs elements along specified axis
  • axis: int。表示用于合并模式的轴。在大多数情况下:concat 为 0,其他模式为 1。
  • name: 此层的名称(可选)。默认值:'Merge'。

合并输出

tflearn.layers.merge_ops.merge_outputs (tensor_list, name='MergeOutputs')

将网络的所有输出连接成单个张量的层。

输入

张量列表 [形状]。

输出

连接后的张量 [张量数, 形状]。

参数

  • tensor_list: Tensor 列表。网络输出。
  • name: str。此层的名称(可选)。

返回值

一个 Tensor