Open Kidleyh opened 9 months ago
Why does line 706 of code look like this? The last two dimensions are opposite. torch.randn( (1,n_component[i],gridSize[mat_id_1] * j,gridSize[mat_id_0] * j, ) But line 731 of code is torch.randn((1, n_component[i], planeSize[0], planeSize[1])
torch.randn( (1,n_component[i],gridSize[mat_id_1] * j,gridSize[mat_id_0] * j, )
torch.randn((1, n_component[i], planeSize[0], planeSize[1])
Why does line 706 of code look like this? The last two dimensions are opposite.
torch.randn( (1,n_component[i],gridSize[mat_id_1] * j,gridSize[mat_id_0] * j, )
But line 731 of code istorch.randn((1, n_component[i], planeSize[0], planeSize[1])