Like I said in the previous post, you need 4 dimensions. You have 3 right now. The
inputs[None, ...]

is basically syntactic sugar for
inputs.unsqueeze(0)

which adds an extra dimension in the 0th position.