PyTorch Conv2d 卷积核参数详解：维度、数据类型、设备与初始化

在 PyTorch 中，torch.nn.functional.conv2d 中的卷积核需要是一个 4 维张量，满足以下要求：\n1. 张量形状为（out_channels, in_channels, kernel_size[0], kernel_size[1]），其中 out_channels 是输出通道数，in_channels 是输入通道数，kernel_size 是卷积核尺寸。\n2. 张量数据类型可以是 torch.float32、torch.float64、torch.float16、torch.bfloat16、torch.int8、torch.int16、torch.int32、torch.int64 或 torch.bool。\n3. 张量需要在合适的设备上，例如 CPU 或 GPU。\n4. 张量值可以是随机初始化的，也可以通过训练得到。