卷积层参数详解

import torch


# n是4 输入维度 图片大小 5*5
input=torch.randn((4,3,5,5))

# 输入维度 输出维度 卷积核大小
y=torch.nn.Conv2d(3,2,3)
print(y(input))

# 得到3*3的图片，输出维度是2，有四组

import tensorflow as tf
import keras

# 可以指定卷积核大小
# 指定输出的维度，这里是2
# 指定图片大小

# n=4 图片大小 5*5 输入通道
input_shape=(4,5,5,3)
x=tf.random.normal(shape=input_shape,mean=0)
# 输出维度2 卷积核大小 3
y=keras.layers.Conv2D(2,3,activation="relu",input_shape=input_shape[1:])
print(y(x))
# 得到 3*3的图片,输出空间维度是2 有四组

(4, 3, 3, 2)