Does PyTorch-Encoding support 3D-network now?

I find following code in encoding.nn.encoding.py
def forward(self, X):
assert(X.size(1) == self.D)
if X.dim() == 3:
# BxDxN
B, D = X.size(0), self.D
X = X.transpose(1, 2).contiguous()
elif X.dim() == 4:
# BxDxHxW
B, D = X.size(0), self.D
X = X.view(B, D, -1).transpose(1, 2).contiguous()
else:
raise RuntimeError('Encoding Layer unknown input dims!')

Does it mean that this version cannot supprot 5D input of 3D network? I quite need to deal with 3D network, if so, can I simply add the X.dim() == 5 and change little to make it work?

Start in encoding.nn.encoding.py at the forward method and inspect how its current 3D and 4D input cases are handled. Determine whether 5D input for 3D networks is intended to be supported; done means providing a clear compatibility answer or implementing and verifying that support without changing existing behavior.