Ok, great. But the interpolate function in PyTorch is very easy to use and actually needs only one line of code.
F.interpolate(tensor, size)

By the way, you should no get any error, because your tensor is 3D already so it works precisely as I shared my code.
As I can see, you provided none or something like that to interpolate because the error says your tensor has {}D dimension!
Finally, I have read a little about albumentations 23 library and it seems it does not support GPU, so you can achieve higher benchmark by using GPU and Torch core.
Good luck