update

2024-04-03 00:06:01 +08:00
parent 7a049ccc5e
commit 5e3e22e25d
2 changed files with 23 additions and 0 deletions
--- a/2024.04.02_convolution_and_pooling/convolution.py
+++ b/2024.04.02_convolution_and_pooling/convolution.py
@@ -0,0 +1,10 @@
+import torch
+input_data = torch.randn(1, 1, 28, 28)
+
+conv_layer = torch.nn.Conv2d(in_channels=1, out_channels=64, kernel_size=3, stride=1, padding=1)
+output_data = conv_layer(input_data)
+print("输出数据的形状：", output_data.shape)
+
+conv_layer = torch.nn.Conv2d(in_channels=1, out_channels=64, kernel_size=3, stride=1, padding=0)
+output_data = conv_layer(input_data)
+print("输出数据的形状：", output_data.shape)