码迷,mamicode.com
首页 > 其他好文 > 详细

TenSorFlow-掩码

时间:2019-01-04 10:34:18      阅读:266      评论:0      收藏:0      [点我收藏+]

标签:import   var   python   ant   on()   port   ali   ini   bin   

#!/usr/bin/env python # -*- coding: utf-8 -*- import tensorflow as tf? x = tf.constant(2) y = tf.constant(20) z = tf.Variable([[6,1,2],[3,4,5],[0,0,0],[8,2,4]]) mask=z[:,0]>0 res = tf.boolean_mask(z,mask) init_assign = tf.global_variables_initializer() with tf.Session() as sess: ? ? sess.run(init_assign) ? ? print sess.run(res)

运行结果:

[[6 1 2]
?[3 4 5]
?[8 2 4]]

TenSorFlow-掩码

标签:import   var   python   ant   on()   port   ali   ini   bin   

原文地址:http://blog.51cto.com/13959448/2338462

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!