From what I understand, torch.randn(layers/depth, rows, columns), which can be seen when executing: torch.randn(2, 3, 3) ==> 2 layers (3x3) matr
torch.randn(layers/depth, rows, columns)
torch.randn(2, 3, 3)