请问，想改成 针对3d数据，该怎么改？ 比如（batch，C， h, w, d），超分到（batch, C, H, W, D）。

[Lornatang]

3D数据改两个地方，第一个是Dataset相关，第二个是模型定义相关, in_channels=4,out_channels=4。