PyTorch torch.adjoint 函数
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)
# 创建实数矩阵
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)
# 创建复数矩阵
A = torch.tensor([[1+1j, 2+2j], [3+3j, 4+4j]])
# 计算共轭转置
result = torch.adjoint(A)
print("原矩阵:")
print(A)
print("共轭转置:")
print(result)

Pytorch torch 参考手册