合并
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
。