10、用pytorch求余弦、反余弦、值裁剪、取对角线元素

东方耀

10、用pytorch求余弦、反余弦、值裁剪、取对角线元素






import torch
import torch.nn.functional as F
import numpy as np
import math


# torch.cos
# torch.acos
# torch.clamp
# torch.diagonal
# 用pytorch求余弦、反余弦、值裁剪、取对角线元素


# wf.transpose(0, 1)[labels]
a = np.array([[0, math.pi, 3], [1, -1, 6]], dtype=np.float32)
t = torch.from_numpy(a)
print(t, t.size())


print("torch.cos(t)=", torch.cos(t))
print("torch.acos(t)=", torch.acos(t))
# clamp 夹紧 夹住 限定范围
print("torch.clamp=", torch.clamp(t, -1, 1))
# diagonal 取主对角线
print("torch.diagonal=", torch.diagonal(t))


# torch.diagonal（输入，偏移= 0，dim1 = 0，dim2 = 1 ） →张量
# input（Tensor）–输入张量。必须至少为二维。
#
# offset（int，optional）–要考虑的对角线。默认值：0（主对角线）。
#
# dim1（int，optional）–取对角线的第一维。默认值：0
#
# dim2（int，optional）–取对角线的第二维。默认值：1。


# 该参数offset控制要考虑的对角线：
#
# 如果offset= 0，则为主要对角线。
#
# 如果offset> 0，则它在主对角线上方。
#
# 如果offset<0，则位于主对角线下方