Python PyTorch view_as_complex用法及代碼示例

本文簡要介紹python語言中 torch.view_as_complex 的用法。

用法:

torch.view_as_complex(input) → Tensor

參數：

input(Tensor) -輸入張量。

將input 的視圖作為複數張量返回。對於 size m1, m2, \dots, mi, 2 的輸入複數張量，此函數返回 size m1, m2, \dots, mi 的新複數張量，其中輸入張量的最後一個維度應表示複數的實部和虛部。

警告

view_as_complex() 僅支持具有 torch.dtype torch.float64 和 torch.float32 的張量。輸入的最後一個維度預計為 size 2。此外，張量的最後一個維度的 stride 必須為 1。所有其他維度的步幅必須是偶數。

例子：

>>> x=torch.randn(4, 2)
>>> x
tensor([[ 1.6116, -0.5772],
        [-1.4606, -0.9120],
        [ 0.0786, -1.7497],
        [-0.6561, -1.6623]])
>>> torch.view_as_complex(x)
tensor([(1.6116-0.5772j), (-1.4606-0.9120j), (0.0786-1.7497j), (-0.6561-1.6623j)])