CNN -1 神经网络-概述2
一:神经网络(operator)
算子:通常指的是神经网络层(neural network layer)中所用的基本数学运算;
1> 线性层(Fully Connected Layer)
也称为全连接层,是最常见的一种层类型。它将输入向量与权重矩阵相乘,然后加上偏置向量得到输出向量。线性层是神经网络中的基本层,其主要功能是将高维输入映射到低维输出;
线性层示例代码
import torch
import torch.nn as nn
# 定义输入向量的维度
input_size = 10
# 定义线性层的输出维度
output_size = 5
# 创建线性层
fc_layer = nn.Linear(input_size, output_size)
# 创建输入向量
input_vector = torch.randn(1, input_size) # 假设输入向量的形状为(1, input_size)
# 进行前向传播计算
output_vector = fc_layer(input_vector)
# 打印输出向量
print(output_vector)
首先导入了torch和torch.nn模块。然后,我们定义了输入向量的维度input_s