|
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,则位于主对角线下方
|
|