keras常见函数使用concatenate等

concatenate

concatnate用于拼接两个tensor,最后得到的是一个tensorflow版的tensor。
官方api文档如下:

tf.keras.backend.concatenate(
    tensors, axis=-1
)
a = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
b = tf.constant([[10, 20, 30], [40, 50, 60], [70, 80, 90]])
tf.keras.backend.concatenate((a, b), axis=-1)

axis指定在具体维度上进行拼接,axis=-1即是在最后一维对tensor进行拼接。
最后输出结果如下:
在这里插入图片描述

  • 注意:keras的tensor和tensorflow里的tensor并不一样,如果在模型中加入了concatnate进行拼接两种类型的tensor(或者都是keras的tensor),那么最后输出的都会是tensorflow的tensor,如果我们构建模型使用的是keras,会在运行中报这样一个错误。
    AttributeError:‘Tensor’ object has no attribute '_keras_history’
  • 这是由于kreas模型构建中不能使用tensorflow中的tensor造成的,那么就需要用如下引入Lamba层将tensorflow的tensor转化为keras的tensor。
from keras.layers import Lambda
# axis为指定拼接的维度
concat_result = Lambda(lambda x: concatenate((x, y), axis=-1))(x)

这样得到的结果就是keras中的tensor,接下来就可以放心的训练了。

  • 2
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值