1. 程式人生 > >tensorflow啟用函式relu()的詳解

tensorflow啟用函式relu()的詳解

tf.nn.relu(features, name = None)

解釋:這個函式的作用是計算啟用函式relu,即max(features, 0)。即將矩陣中每行的非最大值置0。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf

a = tf.constant([-1.0, 2.0])
with tf.Session() as sess:
    b = tf.nn.relu(a)
    print sess.run(b)
以上程式輸出的結果是:[0. 2.]

說明:a中每一個元素與0進行比較,小於0的置為0;大於0的不變。

-----------------------

相關參考: