PyTorch: Wie bekommt man die Form eines Tensors als Liste von int

In numpy, V.shape gibt ein Tupel von int-Werten der Dimensionen V.

In tensorflow V.get_shape().as_list() gibt eine Liste von Ganzzahlen, die die Dimensionen der V.

In pytorch, V.size() gibt ein size-Objekt, aber wie wandle ich es um ints?

InformationsquelleAutor patapouf_ai | 2017-10-19

  1. 14

    Für PyTorch v1.0 und eventuell oben:

    >>> import torch
>>> var = torch.tensor([[1,0], [0,1]])

# Using .size function, returns a torch.Size object.
>>> var.size()
torch.Size([2, 2])
>>> type(var.size())
<class 'torch.Size'>

# Similarly, using .shape
>>> var.shape
torch.Size([2, 2])
>>> type(var.shape)
<class 'torch.Size'>

    Können Sie werfen jede Fackel.Größe Objekt um eine native Python-Liste:

    >>> list(var.size())
[2, 2]
>>> type(list(var.size()))
<class 'list'>

    In PyTorch v0.3 und 0,4:

    Einfach list(var.size()), z.B.:

    >>> import torch
>>> from torch.autograd import Variable
>>> from torch import IntTensor
>>> var = Variable(IntTensor([[1,0],[0,1]]))

>>> var
Variable containing:
 1  0
 0  1
[torch.IntTensor of size 2x2]

>>> var.size()
torch.Size([2, 2])

>>> list(var.size())
[2, 2]
    Anstelle von aufrufen list, ist die Größe der Klasse haben gewisse attribute, die ich direkt zugreifen kann, um die Form in einem Tupel oder Liste form?
    Sie können auf die Größe direkt ohne Umwandlung in eine Liste. Die OP angefordert, um eine Liste damit auch die Art casting.

    InformationsquelleAutor alvas

  2. 4

    Wenn Sie ein fan von NumPyish syntax, dann gibt es tensor.shape.

    In [3]: ar = torch.rand(3, 3)

In [4]: ar.shape
Out[4]: torch.Size([3, 3])

# method-1
In [7]: list(ar.shape)
Out[7]: [3, 3]

# method-2
In [8]: [*ar.shape]
Out[8]: [3, 3]

# method-3
In [9]: [*ar.size()]
Out[9]: [3, 3]

    P. S.: Beachten Sie, dass tensor.shape ist ein alias für tensor.size(), obwohl tensor.shape ist ein Attribut der tensor in Frage, in der Erwägung, dass tensor.size() ist eine Funktion.

    Welcher Teil der code funktioniert nur auf einem GPU-Maschine? tensor.shape?

    InformationsquelleAutor kmario23

Schreibe einen Kommentar