Berbagi teknologi

한어Русский языкEnglishFrançaisIndonesianSanskrit日本語DeutschPortuguêsΕλληνικάespañolItalianoSuomalainenLatina

1: Jaringan saraf (operator)

Operator: biasanya mengacu pada operasi matematika dasar yang digunakan pada lapisan jaringan saraf;

1> Lapisan linier (Lapisan Terhubung Sepenuhnya)

Juga disebut lapisan terhubung penuh, ini adalah jenis lapisan yang paling umum. Ini mengalikan vektor masukan dengan matriks bobot dan kemudian menambahkan vektor bias untuk mendapatkan vektor keluaran. Lapisan linier adalah lapisan dasar dalam jaringan saraf, dan fungsi utamanya adalah memetakan masukan berdimensi tinggi ke keluaran berdimensi rendah;

Kode sampel lapisan linier

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)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

Pertama, modul torch dan torch.nn diimpor. Kemudian, kita mendefinisikan dimensi input_size dari vektor input dan dimensi output output_size dari lapisan linier.

Selanjutnya, kita membuat objek lapisan linier bernama fc_layer menggunakan kelas nn.Linear. Parameter pertama kelas ini adalah dimensi masukan, dan parameter kedua adalah dimensi keluaran.

Kami kemudian membuat vektor input acak input_vector dan meneruskannya ke lapisan linier untuk perhitungan propagasi maju.Terakhir, kami mencetak vektor keluaran