Skip to content

Instantly share code, notes, and snippets.

@thomelane
Last active September 21, 2018 06:37
Show Gist options
  • Save thomelane/a2d89209a9cea0e644f5ccab945824c8 to your computer and use it in GitHub Desktop.
Save thomelane/a2d89209a9cea0e644f5ccab945824c8 to your computer and use it in GitHub Desktop.
Convolution layouts on Medium
Convolution Input Layout Kernel Layout Output Layout
1D (batch_size,in_channels,width) (channels,in_channels,width) (batch_size,channels,width)
2D (batch_size,in_channels,height,width) (channels,in_channels,height,width) (batch_size,channels,height,width)
3D (batch_size,in_channels,depth,height,width) (channels,in_channels,depth,height,width) (batch_size,channels,depth,height,width)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment