卷积神经网络中如何使用余弦函数作为激活函数?

时间:2019-01-08 07:59:32

标签: python tensorflow

使用tf.variable_scope('layer1-conv1'):

    conv1_weights = tf.get_variable("weight",[3,3,3,32],initializer=tf.truncated_normal_initializer(stddev=0.1))
    conv1_biases = tf.get_variable("bias", [32], initializer=tf.constant_initializer(0.0))
    conv1 = tf.nn.conv2d(input_tensor, conv1_weights, strides=[1, 1, 1, 1], padding='SAME')
    relu1 = tf.nn.relu(tf.nn.bias_add(conv1, conv1_biases))

以上述层为例,如何用sin(x)替换tf.nn.relu? tf.nn.bias_add(conv1,conv1_biases)是张量,但是math.sin()的参数是数字。希望您能对我有所帮助。谢谢!!!

1 个答案:

答案 0 :(得分:1)

使用tf.math.sin作为激​​活函数,因为它将在张量上逐元素计算该函数。