常用的pytorch函数
torch.pow(input, exponent, out=None)对输入input的按元素求exponent次幂值,并返回结果张量。 幂值exponent可以为单一float数或者与input相同元素数的张量。>>> exp = torch.arange(1, 5)>>> a = torch.arange(1, 5)>>&...
·
torch.pow(input, exponent, out=None)
对输入input的按元素求exponent次幂值,并返回结果张量。 幂值exponent 可以为单一 float 数或者与input相同元素数的张量。
>>> exp = torch.arange(1, 5)
>>> a = torch.arange(1, 5)
>>> a
1
2
3
4 [torch.FloatTensor of size 4]
>>> exp
1
2
3
4 [torch.FloatTensor of size 4]
>>> torch.pow(a, exp)
1
4
27
256 [torch.FloatTensor of size 4]
torch.matmul(mat1, mat2, out=None) → Tensor
对矩阵mat1和mat2进行相乘。 如果mat1 是一个n×m张量,mat2 是一个 m×p 张量,将会输出一个 n×p 张量out。
参数 :
mat1 (Tensor) – 第一个相乘矩阵
mat2 (Tensor) – 第二个相乘矩阵
out (Tensor, optional) – 输出张量
持续更新中······
更多推荐




所有评论(0)