Question

火炬文档中 torch.empty_like 的描述说：

<块引用>

torch.empty_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format) → Tensor

返回与输入大小相同的未初始化张量。 torch.empty_like(input) 等价于 torch.empty(input.size(), dtype=input.dtype, layout=input.layout, device=input.device)。

参数
input (Tensor) – 输入的大小将决定输出张量的大小。

我要做的是：

>>> torch.empty(3,4)
tensor([[-1.8597e+15,  4.5657e-41, -1.8597e+15,  4.5657e-41],
        [ 4.4842e-44,  0.0000e+00,  8.9683e-44,  0.0000e+00],
        [ 0.0000e+00,  0.0000e+00,  0.0000e+00,  0.0000e+00]])

>>> c1
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])

>>> torch.empty_like(c1)
tensor([[139942262173040,  93851872482144,               1,               0],
        [              0,               0,  93851872492496,               0],
        [              0,               0,               0,               0]])

>>> d
tensor([[0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [0., 0., 0., 0.]])

>>> torch.empty_like(d)
tensor([[-8.6092e-25,  3.0620e-41,  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]])

torch.empty_like 返回的张量似乎取决于输入值，与文档中的描述相反。有人能解释一下吗？

Answer 1

文档描述是正确的。我不确定您是否对 torch.empty_like 在不同的调用中返回不同的输出感到困惑，但是您可以通过调用例如 torch.empty 来看到这也是 torch.empty((2,3), dtype=torch.int64) 的行为torch.empty_like 多次。

注意 dtype 确实取决于输入的 QueueHandler（但不是其特定值）。

torch.empty_like() 是否取决于输入值以及输入大小？

1 个答案: