现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.adjoint 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

torch.adjoint 是 PyTorch 中用于计算矩阵的共轭转置(adjoint)的函数。对于实数矩阵,它等价于转置;对于复数矩阵,它返回转置并取每个元素的共轭。

函数定义

torch.adjoint(input)

参数:

  • input (Tensor): 输入张量。

返回值:

  • torch.Tensor: 返回输入张量的共轭转置。

使用示例

实例 - 实数矩阵

import torch

# 创建实数矩阵
A = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 计算共轭转置
result = torch.adjoint(A)

print("原矩阵:")
print(A)
print("形状:", A.shape)
print("共轭转置:")
print(result)
print("形状:", result.shape)

输出结果为:

原矩阵:
tensor([[1, 2, 3],
        [4, 5, 6]])
形状: torch.Size([2, 3])
共轭转置:
tensor([[1, 4],
        [2, 5],
        [3, 6]])
形状: torch.Size([3, 2])

实例 - 复数矩阵

import torch

# 创建复数矩阵
A = torch.tensor([[1+1j, 2+2j], [3+3j, 4+4j]])

# 计算共轭转置
result = torch.adjoint(A)

print("原矩阵:")
print(A)
print("共轭转置:")
print(result)

Pytorch torch 参考手册 Pytorch torch 参考手册