Tensorflow之tf.split()

https://blog.csdn.net/weixin_34613450/article/details/82725036
https://blog.csdn.net/kdongyi/article/details/82910632

split(
    value,
    num_or_size_splits,
    axis=0,
    num=None,
    name='split'
)		

输入:
value:输入的tensor
num_or_size_splits:如果是个整数n,就将输入的tensor分为n个子tensor。如果是个tensor T,就将输入的tensor分为len(T)个子tensor
axis:默认为0,指明在哪个维度上分

举例:

import tensorflow as tf
import numpy as np

value = np.array([[1,2,3],
                  [4,5,6]])

#axis = 0,表示按行切分,切分的子tensor列数不变
#将value按行切分成两份,第一份0行3列,第2份2行3列
split1,split2 = tf.split(value,[0,2],0)

#axis = 1,表示按列切分,切分的子tensor行数不变
#将value按列切分乘三份,第一份2行1列,第二份2行0列,第三份2行2列
split3,split4,split5 = tf.split(value, [1, 0, 2], 1)

with tf.Session() as sess:
    print("split1:\n")
    print(sess.run(split1))
    print("split2:\n")
    print(sess.run(split2))
    print("split3:\n")
    print(sess.run(split3))
    print("split4:\n")
    print(sess.run(split4))
    print("split5:\n")
    print(sess.run(split5))

结果如下:

split1:

[]
split2:

[[1 2 3]
 [4 5 6]]
split3:

[[1]
 [4]]
split4:

[]
split5:

[[2 3]
 [5 6]]
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
import time import tensorflow.compat.v1 as tf tf.disable_v2_behavior() from tensorflow.examples.tutorials.mnist import input_data import mnist_inference import mnist_train tf.compat.v1.reset_default_graph() EVAL_INTERVAL_SECS = 10 def evaluate(mnist): with tf.Graph().as_default() as g: #定义输入与输出的格式 x = tf.compat.v1.placeholder(tf.float32, [None, mnist_inference.INPUT_NODE], name='x-input') y_ = tf.compat.v1.placeholder(tf.float32, [None, mnist_inference.OUTPUT_NODE], name='y-input') validate_feed = {x: mnist.validation.images, y_: mnist.validation.labels} #直接调用封装好的函数来计算前向传播的结果 y = mnist_inference.inference(x, None) #计算正确率 correcgt_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(y_, 1)) accuracy = tf.reduce_mean(tf.cast(correcgt_prediction, tf.float32)) #通过变量重命名的方式加载模型 variable_averages = tf.train.ExponentialMovingAverage(0.99) variable_to_restore = variable_averages.variables_to_restore() saver = tf.train.Saver(variable_to_restore) #每隔10秒调用一次计算正确率的过程以检测训练过程中正确率的变化 while True: with tf.compat.v1.Session() as sess: ckpt = tf.train.get_checkpoint_state(minist_train.MODEL_SAVE_PATH) if ckpt and ckpt.model_checkpoint_path: #load the model saver.restore(sess, ckpt.model_checkpoint_path) global_step = ckpt.model_checkpoint_path.split('/')[-1].split('-')[-1] accuracy_score = sess.run(accuracy, feed_dict=validate_feed) print("After %s training steps, validation accuracy = %g" % (global_step, accuracy_score)) else: print('No checkpoint file found') return time.sleep(EVAL_INTERVAL_SECS) def main(argv=None): mnist = input_data.read_data_sets(r"D:\Anaconda123\Lib\site-packages\tensorboard\mnist", one_hot=True) evaluate(mnist) if __name__ == '__main__': tf.compat.v1.app.run()对代码进行改进
最新发布
05-26
以下是对代码的改进建议: 1. 在代码开头添加注释,简要说明代码功能和使用方法。 2. 将导入模块的语句放在代码开头。 3. 将模型保存路径和评估时间间隔定义为常量,并使用有意义的变量名。 4. 将计算正确率和加载模型的过程封装为函数。 5. 在主函数中调用评估函数。 改进后的代码如下: ``` # 该代码实现了使用已训练好的模型对 MNIST 数据集进行评估 import time import tensorflow.compat.v1 as tf from tensorflow.examples.tutorials.mnist import input_data import mnist_inference import mnist_train # 定义常量 MODEL_SAVE_PATH = 'model/' EVAL_INTERVAL_SECS = 10 def evaluate(mnist): """ 计算模型在验证集上的正确率 """ with tf.Graph().as_default() as g: # 定义输入和输出格式 x = tf.placeholder(tf.float32, [None, mnist_inference.INPUT_NODE], name='x-input') y_ = tf.placeholder(tf.float32, [None, mnist_inference.OUTPUT_NODE], name='y-input') # 直接调用封装好的函数计算前向传播结果 y = mnist_inference.inference(x, None) # 计算正确率 correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(y_, 1)) accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32)) # 加载模型 variable_averages = tf.train.ExponentialMovingAverage(mnist_train.MOVING_AVERAGE_DECAY) variables_to_restore = variable_averages.variables_to_restore() saver = tf.train.Saver(variables_to_restore) # 在验证集上计算正确率 with tf.Session() as sess: ckpt = tf.train.get_checkpoint_state(MODEL_SAVE_PATH) if ckpt and ckpt.model_checkpoint_path: saver.restore(sess, ckpt.model_checkpoint_path) global_step = ckpt.model_checkpoint_path.split('/')[-1].split('-')[-1] accuracy_score = sess.run(accuracy, feed_dict={x: mnist.validation.images, y_: mnist.validation.labels}) print("After %s training steps, validation accuracy = %g" % (global_step, accuracy_score)) else: print('No checkpoint file found') def main(argv=None): # 读取数据集 mnist = input_data.read_data_sets('MNIST_data', one_hot=True) # 每隔一定时间评估模型在验证集上的正确率 while True: evaluate(mnist) time.sleep(EVAL_INTERVAL_SECS) if __name__ == '__main__': tf.app.run() ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值