torch.empty()
时间: 2023-08-26 19:03:43 浏览: 180
PyTorch中torch.tensor与torch.Tensor的区别详解
torch.empty() 是一个 PyTorch 函数,用于创建一个指定大小的未初始化的张量(tensor)。未初始化的张量不会被清零,它们的值是未知的,可能包含任意值。该函数的语法如下:
```python
torch.empty(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor
```
其中,*size 是一个可变参数,用于指定张量的大小。out 是一个可选参数,用于指定输出张量。dtype 是一个可选参数,用于指定张量的数据类型。layout 是一个可选参数,用于指定张量的布局。device 是一个可选参数,用于指定张量所在的设备。requires_grad 是一个可选参数,用于指定张量是否需要计算梯度。
例如,创建一个大小为 2x3x4 的未初始化的张量可以使用以下代码:
```python
import torch
x = torch.empty(2, 3, 4)
print(x)
```
输出结果如下:
```
tensor([[[4.6439e-36, 0.0000e+00, 2.8026e-45, 0.0000e+00],
[ nan, 0.0000e+00, 0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00]],
[[0.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00],
[0.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00]]])
```
注意,该张量的值为未知值,可能为任意值。
阅读全文