• 从softmax分类问题看神经网络的核心


    def net(X):
        return softmax(torch.matmul(X.reshape((-1, W.shape[0])), W) + b)

    神经网络的核心是矩阵相乘,torch.matmul(X.reshape((-1, W.shape[0])), W) + b,  这里是,

    torch.matmul(X.reshape((-1, W.shape[0])), W) 做了x和w的矩阵点积相乘,x是输入的图像矩阵,灰度图是(h,w),彩色是(h,w,c);W.shape[0],w是二维,

    这个表达式来自PyTorch框架,用于执行一个矩阵乘法操作后加上偏置(bias)的操作,是神经网络中全连接层(也称作密集连接层或线性层)计算的核心部分。下面是对这个表达式的逐步解释:

    1. **X.reshape((-1, W.shape[0]))**:
       - `X` 是输入数据,通常是一个二维张量(矩阵),形状可能是`(batch_size, feature_size)`,其中`batch_size`是批量大小,`feature_size`是输入特征的数量。
       - `.reshape((-1, W.shape[0]))` 这个操作将`X`调整形状,使得它能与权重矩阵`W`进行矩阵乘法。`-1`是一个特殊的值,表示这一维度的大小将自动计算以保持元素总数不变。`W.shape[0]`表示权重矩阵`W`的列数,意味着我们将`X`调整为`(batch_size * height * width, out_features)`的形状,其中`out_features`是权重矩阵的列数,对应于输出层的神经元数量。这样的重塑是为了确保输入数据的每个样本都能与权重矩阵正确相乘。

    2. **torch.matmul(..., W)**:
       - `torch.matmul`函数执行矩阵乘法。在这个情况下,它将重塑后的输入数据与权重矩阵`W`相乘。如果重塑后的`X`形状为`(batch_size * height * width, in_features)`,而`W`的形状为`(in_features, out_features)`,那么结果将会是一个形状为`(batch_size * height * width, out_features)`的张量,表示每个输入样本通过线性变换后得到的输出特征向量。

    3. **... + b**:
       - 最后,将上述矩阵乘法的结果与偏置项`b`相加。偏置`b`通常是一个一维张量,形状为`(out_features,)`,对应于输出层每一个神经元的偏置项。相加操作为每个输出特征添加了一个常数项,增加了模型的表达能力,有助于解决数据可能存在的偏移问题。

    综上所述,这个表达式完成了神经网络中一个典型全连接层的前向传播计算:首先对输入数据进行适当的形状调整以匹配权重矩阵的维度要求,然后执行与权重的矩阵乘法,最后加上偏置项,得到该层的输出。这是构建神经网络模型时非常基础且重要的一部分

    • 权重(W)的维度:权重的维度依赖于它在模型中的位置和作用。以卷积神经网络(CNN)为例:

      • 全连接层权重: 如果在模型中有一个全连接层紧接在卷积层之后,该全连接层的权重会是一个二维矩阵,其维度取决于前一层的输出数量(或经过展平操作后的大小)和该全连接层的节点数。
      • 卷积层权重: 通常是一个四维张量,形状为(out_channels, in_channels, kernel_height, kernel_width)。这里,out_channels是输出特征图的数量,in_channels是输入图像的通道数(例如,对于第一层可能为1(灰度)或3(RGB)),kernel_heightkernel_width定义了卷积核的大小。
  • 相关阅读:
    使用 Web HID API 在浏览器中进行HID设备交互(纯前端)
    架构基本概念和架构本质
    c#面试编程题
    JavaScript学习--Day05
    优秀案例 | 虚拟人“唯思”主持2023世界青年科学家峰会
    初识docker
    进程间的通信终章之【消息队列,信号量,共享内存】
    Android项目使用gradle配置新旧方式对比
    django ninja get not allowed 能用 put delete
    MySQL:索引
  • 原文地址:https://blog.csdn.net/zw_van/article/details/139552286