torch.full¶
- torch.full(size, fill_value, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) Tensor¶
建立一個大小為
size、填充值為fill_value的 Tensor。Tensor 的 dtype 從fill_value推斷。- 引數
size (int...) – 定義輸出 Tensor 形狀的整數列表、元組或
torch.Size。fill_value (Scalar) – 用於填充輸出 Tensor 的值。
- 關鍵字引數
out (Tensor, optional) – 輸出 Tensor。
dtype (
torch.dtype, optional) – 返回 Tensor 的期望資料型別。預設值：如果為 `None`，則使用全域性預設值（參見torch.set_default_dtype()）。layout (
torch.layout, optional) – 返回 Tensor 的期望佈局。預設值：torch.strided。device (
torch.device, optional) – 返回 Tensor 的期望裝置。預設值：如果為 `None`，則對預設 Tensor 型別使用當前裝置（參見torch.set_default_device()）。對於 CPU Tensor 型別，device將是 CPU；對於 CUDA Tensor 型別，則將是當前的 CUDA 裝置。requires_grad (bool, optional) – 如果 autograd 應該記錄返回 Tensor 上的操作。預設值：`False`。
示例
>>> torch.full((2, 3), 3.141592) tensor([[ 3.1416, 3.1416, 3.1416], [ 3.1416, 3.1416, 3.1416]])
