python使用torch随机初始化三维张量

在PyTorch中，你可以使用`torch.nn.init`模块中的函数来随机初始化三维张量（通常代表神经网络中的权重或偏置）。其中，`torch.zeros_()` 或 `torch.randn_()` 可以用来创建全零或随机分布的张量，而`torch.Tensor.new_*` 方法可以创建新的张量并应用初始化。

例如，如果你想要创建一个形状为`(batch_size, channels, height, width)`的三维张量，初始值为零，你可以这样做：

import torch
from torch.nn import init

# 定义维度
batch_size = 10
channels = 64
height = 28
width = 28

# 创建三维张量
tensor = torch.empty(batch_size, channels, height, width)
init.zeros_(tensor)  # 初始化全部为0

# 或者如果需要随机初始化，可以这样
tensor = tensor.new_zeros(batch_size, channels, height, width)

# 如果想要标准正态分布，替换 zeros_() 为 normal_()
init.normal_(tensor)

在这里，`new_zeros_()` 和 `new_normal_()` 都会在创建新张量的同时进行初始化操作，省去了先创建空张量再填充数据的步骤。