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) – 输出张量

持续更新中······

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐