tensorflow
King的王国
AI干货分享
展开
-
tensorflow框架----ckpt转pb模型
1、ckpt转pb import tensorflow as tf def freeze_graph(input_checkpoint, output_graph): # 指定输出的节点名称,该节点名称必须是原模型中存在的节点(重要!!!!!!!!!,如何找到该节点名称,见下文) output_node_names = ['attn_cell_1/transpose_1'] saver = tf.train.import_meta_graph(input_checkpoint + '原创 2021-03-16 11:11:40 · 224 阅读 · 0 评论 -
极简TensorFlow学习教程-----TensorFow中Tensor与Numpy相互转换
1、Numpy转Tensor TensorFlow 通过 convert_to_tensor() 函数将Numpy转换为Tensor,代码如下: import tensorflow as tf import numpy as np # 创建ndarray array = np.array([1, 2, 3, 4, 5], np.float32) print(array) #[1. 2. 3. 4. 5.] # 将ndarray转化为tensor t = tf.convert_to_tensor(array,原创 2020-05-27 13:43:34 · 3148 阅读 · 1 评论