目录

• 2D Convolution
• Kernel size
• Padding & Stride
• Channels
• For instance
• LeNet-5
• Pyramid Architecture
• layers.Conv2D
• weight & bias
• nn.conv2d
• Gradient?
  • For instance

2D Convolution

Kernel size

• 矩阵卷积

Padding & Stride

• 步长2

Channels

For instance

• x: [b,28,28,3]
• one k: [3,3,3]
• multi-k: [16,3,3,3]
• stride: 1
• padding: [1,1,1,1]
• bias: [16]
• out: [b,28,28,16]

LeNet-5

Pyramid Architecture

• 从底层的边缘颜色到高层抽象的概念（轮子、车窗）

layers.Conv2D

import tensorflow as tf
from tensorflow.keras import layers

x = tf.random.normal([1, 32, 32, 3])

# padding='valid'：输入和输出维度不同
layer = layers.Conv2D(4, kernel_size=5, strides=1, padding='valid')
out = layer(x)
out.shape

TensorShape([1, 28, 28, 4])

# padding='same'：输入和输出维度相同
layer = layers.Conv2D(4, kernel_size=5, strides=1, padding='same')
out = layer(x)
out.shape

TensorShape([1, 32, 32, 4])

layer = layers.Conv2D(4, kernel_size=5, strides=2, padding='same')
out = layer(x)
out.shape

TensorShape([1, 16, 16, 4])

layer.call(x).shape

TensorShape([1, 16, 16, 4])

weight & bias

layer = layers.Conv2D(4, kernel_size=5, strides=2, padding='same')
out = layer(x)
out.shape

TensorShape([1, 16, 16, 4])

# 5，5--》size，3--》通道数，4--》核数量
layer.kernel.shape

TensorShape([5, 5, 3, 4])

layer.bias

<tf.Variable 'conv2d_11/bias:0' shape=(4,) dtype=float32, numpy=array([0., 0., 0., 0.], dtype=float32)>

nn.conv2d

w = tf.random.normal([5, 5, 3, 4])
b = tf.zeros([4])
x.shape

TensorShape([1, 32, 32, 3])

out = tf.nn.conv2d(x, w, strides=1, padding='VALID')
out.shape

TensorShape([1, 28, 28, 4])

out = out + b
out.shape

TensorShape([1, 28, 28, 4])

out = tf.nn.conv2d(x, w, strides=2, padding='VALID')
out.shape

TensorShape([1, 14, 14, 4])

Gradient?

∂Loss∂w$$\frac{\mathrm{\partial }Loss}{\mathrm{\partial }w}$$

For instance

标签：layer,卷积,28,shape,padding,神经网络,TensorShape,out
来源： https://www.cnblogs.com/abdm-989/p/14123375.html