Which dimension does (SR*b*b) describe?
If it’s the channel dimension, you could create a conv layer with in_channels=SR*b*b and out_channels=b*b.
However, based on the provided dimensions I’m a bit confused what type of input you are dealing with.
Could you describe your use case a bit?