模型输入之前的torch.randn(1, 1, 32, 32)中各个参数具体什么含义

对于二维图像：
在调用模型的时候，我们会经常看到下图中的代码，除了模型之外，都会有一个输入X，但是并不知道什么含义，下面来解释一下：

• 如图所示
  
  torch.randn[8, 3, 244, 244]，[batch, channel, height, width]，表示batch_size=8， 3通道（灰度图像为1），图片尺寸：224x224
  如果给的是torch.randn[1, 1, 32, 32]表示batch_size=1， 1通道（灰度图像），图片尺寸：32x32

对于三维图像：

test_data = torch.randn(batch_size, channels, depth, height, width)，用该行代码测试模型能够正常运行。