Number of channels per group ({1}, specified by the size of the 'C' dimension of the weights) times the number of groups ({2}, specified by the size of the second 'U' dimension of the weights) must be equal to the size of the 'C' dimension of the input data ({0})