full_grid = torch.cat([trans_grid, grid_z.unsqueeze(-1)], -1) 5D情况下grid.shape=(b,d,h,w,3),那么单独提取出grid[i,j,k,l]是否等于(z,y,x)?感觉很奇怪的定义。
full_grid = torch.cat([trans_grid, grid_z.unsqueeze(-1)], -1)
5D情况下grid.shape=(b,d,h,w,3),那么单独提取出grid[i,j,k,l]是否等于(z,y,x)?感觉很奇怪的定义。