把tf.map_fn替换成tf.where实现
举例：
把label_tensor = tf.map_fn(lambda x: tf.cond(x >= 1, lambda: 1, lambda: 0), label_tensor) 改成label_tensor = tf.where(label_tensor >= 1, tf.ones_like(label_tensor), tf.zeros_like(label_tensor))
TensorFlow（1.15版本），tf.map_fn速度很慢
原创TechOnly 博主文章分类：TensorFlow ©著作权
文章标签 tensorflow 文章分类 Python 后端开发
©著作权归作者所有：来自51CTO博客作者TechOnly的原创作品，请联系作者获取转载授权，否则将追究法律责任
提问和评论都可以，用心的回复会被更多人看到
评论
发布评论
相关文章
-
Exchange邮件批量删除工具1.0版本
exchange批量删除邮件
Outlook 快速查找 邮件系统 -
如何在Rocky 9.4版本中安装Mongodb
如何在Rocky 9.4版本中安装和配置Mongodb
mongodb linux SQL -
【Python】Tensorflow-cpu-1.15版本环境准备
python环境准备安装Anaconda并配置环境对应的python版本为：3.7.4：pip配置在对应用户目录下，本机示例如下
python3.7.4 tensorflow tensorflow-cpu win10 1.15 -
奇怪的TensorFlow，tf1.5和tf1.15，都报错
1.15不能用，1.5却可以
tensorflow 1.5 1.15 -
tensorflow1版本和2版本语句兼容
tensorflow1版本和2版本语句兼容
tensorflow -
kubernetes二进制部署1.15版本（1）
k8s集群二进制1.15版本脚本半自动化部署
k8s1.15 etcd部署 flanneld网络 -
tensorflow 1.15 KL loss 代码
from tensorflow.python.keras.utils import losses_ut
git tensorflow python -
tensorflow2版本哪个最好用 tensorflow哪个版本最流行
2019年下半年， tensorflow更新到了2.0版本，对Keras API进行了兼容，使用起来更加方便，可以快速搭建神经网络进行模型计算。最近几年陆续也利用工作之余，学习过一些机器学习和神经网络的书籍，许多内容没有吃透，因此准备整理下之前的材料，作为学习笔记，也和大家一起沟通学习。 &n
tensorflow2版本哪个最好用 tensorflow AI tensorflow2 anaconda
