softmax layer 代码

以下是使用Python实现softmax layer的示例代码：

import numpy as np

def softmax(x):
    """
    计算softmax函数的值
    
    Args:
        x: numpy数组，形状为(1, n)，n为向量的维度
        
    Returns:
        s: numpy数组，形状为(1, n)，softmax函数的值
    """
    exp_x = np.exp(x)
    s = exp_x / np.sum(exp_x, axis=1, keepdims=True)
    return s

该函数接收一个形状为(1, n)的numpy数组，其中n为向量的维度，返回一个形状也为(1, n)的numpy数组，表示softmax函数的值。其中，np.exp(x)计算每个元素的指数值，np.sum(exp_x, axis=1, keepdims=True)计算所有元素的和，并将结果保留为二维数组。最后将每个元素除以总和，得到softmax函数的值。