关于p_y_1的疑问

来源:2-6 数据处理与模型图构建(2)

技术进步官

2020-04-09

老师,我看见代码里有predict=p_y_1>0.5,p_y_1的shape不是一个集合吗,难道是可以直接与数字比大小再生成一个新的predict集合?

写回答

1回答

正十七

2020-04-15

对,是的,会生成一个bool的tensor。

即输入a = tf.tensor([0.6, 0.7, 0.1, 0.2]) 

a > 0.5 => tf.tensor([true, true, false, false])

1
0

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程