DEV Community
·
PyTorch中的sub函数
💡
原文英文，约500词，阅读约需2分钟。
📝
内容提要
本文介绍了PyTorch中的sub()函数，该函数用于执行张量减法，支持标量和张量输入，并可选用参数alpha进行缩放。示例展示了不同的减法操作。
🎯
关键要点
- sub()函数用于执行张量减法，支持标量和张量输入。
- sub()函数的第一个参数是输入张量或标量，第二个参数是要减去的张量或标量，第三个参数alpha是可选的缩放因子。
- subtract()是sub()的别名。
- 示例展示了不同的减法操作，包括使用不同的alpha值进行缩放。
- 可以对0D或多维张量进行减法操作，结果也是0D或多维张量。
➡️
