Unfortunately, you haven’t properly formatted the code and even after copying and fixing the code locally, the code is still not executable since BasicBlock3x3 is undefined.
As previously described, check the input shape and make sure it’s a 3D tensor with the expected shapes.