python - 如何在卷积神经网络中使用余弦函数作为激活函数？

Question

与 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))

以上层为例，如何将tf.nn.relu替换为sin(x)？ tf.nn.bias_add(conv1, conv1_biases) 是一个张量，但是 math.sin() 的参数是一个数字。我希望你能帮助我。谢谢!!!

Answer 1

使用tf.math.sin作为您的激活函数，因为它将在您的张量上按元素计算函数。