python的torch如何加入指数幂
时间: 2024-02-23 17:59:57 浏览: 123
python torch.utils.data.DataLoader使用方法
5星 · 资源好评率100%
在PyTorch中,可以使用torch.exp()函数来计算输入张量的指数幂。例如,如果你有一个名为x的张量,并且想要计算e的x次幂,则可以使用下面的代码:
```
import torch
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.exp(x)
print(y) # 输出:tensor([ 2.7183, 7.3891, 20.0855])
```
在上面的代码中,我们首先创建了一个包含三个元素的张量x,然后使用torch.exp()函数计算了x的指数幂,并将结果存储在y中。最后,我们打印了y的值,得到了一个包含三个元素的张量,其中每个元素都是x对应元素的指数幂。
阅读全文