Hi,
What is the stack trace that comes with the error?
I think the problem is that you try to do batchnorm with a Tensor of size [1, 16]. So that has batch size of 1 and 16 channels.
But then there is only a single value and so the standard deviation computed by the batchnorm will lead to infinites when you divide by it.