DEV Community
·
PyTorch 中的 exp() 和 exp2() 函数
💡
原文英文，约400词，阅读约需2分钟。
📝
内容提要
本文介绍了torch中的exp()和exp2()函数的用法。exp()函数可以计算输入张量中每个元素的指数函数值，返回一个浮点型张量。exp2()函数可以计算输入张量中每个元素以2为底的指数函数值，返回一个浮点型张量。两个函数都可以使用torch或张量作为输入，并且可以使用out参数指定输出张量。
🎯
关键要点
- 本文介绍了torch中的exp()和exp2()函数的用法。
- exp()函数计算输入张量中每个元素的指数函数值，返回浮点型张量。
- exp2()函数计算输入张量中每个元素以2为底的指数函数值，返回浮点型张量。
- 两个函数都可以使用torch或张量作为输入，并且可以使用out参数指定输出张量。
- exp()函数的输入参数为张量，返回值为浮点型张量，除非输入张量为复数张量。
- exp2()函数是torch.special.exp2()的别名，其公式为y = 2^x。
- 示例代码展示了如何使用exp()和exp2()函数处理不同类型的张量。
➡️
