torch.full函数介绍

torch.full 是 PyTorch 中用于创建一个具有指定形状、填充值和数据类型的张量的函数。它非常适用于需要初始化特定数值的张量的情况，比如将所有元素填充为一个常量值。

函数定义

torch.full(size, fill_value, *, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor

参数说明

• size: tuple，指定要创建张量的形状（如 (2, 3) 表示 2 行 3 列的矩阵）。
• fill_value: float 或 int，用来填充张量的数值。
• dtype: torch.dtype，张量的数据类型（如 torch.float、torch.int 等）。默认根据 fill_value 推断。
• layout: torch.layout，张量的布局，默认是 torch.strided，表示标准的内存布局。
• device: torch.device，指定创建张量的设备（如 "cpu" 或 "cuda"）。
• requi