TensorFlow和Keras常用方法(避坑)

TensorFlow

在TensorFlow中,除法运算:

  • 1.tensor除法会使结果的精度高一级,可能会导致后面计算类型不匹配,如float32 / float32 = float64。
  • 2.除法需要分子分母同类型,否则报错。
    产生类似错误提示如下:
    -1.TypeError: x and y must have the same dtype, got tf.float32 != tf.int32
    -2.TypeError: Input ‘y’ of ‘Mul’ Op has type float32 that does not match type float64 of argument ‘x’.
    -3.ValueError: Tensor conversion requested dtype float64 for Tensor with dtype float32: ‘Tensor(“Sum:0”, shape=(), dtype=float32)’
    -4.ValueError: Incompatible type conversion requested to type ‘int32’ for variable of type ‘float32_ref’

    解决办法:
    tf.cast(a, tf.float32) # 转换成同类型即可
    

tf.boolean_mask
K.gather
K.argmax
K.max

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值