本文共 2231 字，大约阅读时间需要 7 分钟。

PyTorch ������������������

������������������������������������������������������������������������������������������������������������ PyTorch ������������������������������������Conv1d���������������������Conv2d���������������������������������������������������������������������������������������

���������������Conv1d���

������������������������������������������������������������ element-wise ������������������������������������������������������������������������������

• ���������������(N, C_in, L)
• ���������������(N, C_out, L_out)

������������������������ o u t(N_i, C_out_j) = bias(C_out_j) + ���_{k=0}^{C_in-1} weight(C_out_j, k) ��� input(N_i, k)

���������N_i���������������������������������C_out_j���C_in������������������������������������������������������ ������ element-wise ���������

���������������������

• in_channels���������������������
• out_channels���������������������
• kernel_size���������������������
• stride��������������������������� 1���
• padding��������������������������������������� 0���
• dilation������������������������������ 1���
• groups������������������������������������������������������ 1���
• bias��������������������������������� True���

���������������Conv2d���

��������������������������������������������������������� ������������������������������������������������������������

• ���������������(N, C_in, H_in, W_in)
• ���������������(N, C_out, H_out, W_out)

������������������ H_out = floor((H_in + 2 * padding[0] - dilation[0] * (kernel_size[0] - 1) - 1) / stride[0] + 1) W_out = floor((W_in + 2 * padding[1] - dilation[1] * (kernel_size[1] - 1) - 1) / stride[1] + 1)

���������������������������������������������������������������������������������������������������������������������������������������������������

���������������������

• kernel_size��������������������������������������������� 3x3��������������������� shape���3,5���������
• stride������������������������������������������
• padding��������������������������������� 0 ��������������������� 0���
• dilation���������������������������������������������������������������
• groups������������������������������������������������������

������������������������������������������������������������������������������������������������������