import torch
torch.sub(input, other, *, alpha=1, out=None)
torch.sub() 函数用于两个 tensor 间的相减
参数
input: 输入的被减数,格式为tensor格式
other:输入的减数
alpha:与上面other参数搭配使用,用来与other相乘,当使用torch.sub()函数时不指定alpha的值时,alpha默认为1
out: 指定torch.sub()输出值被赋给的变量,可不指定
import torch
torch.sub(input, other, *, alpha=1, out=None)
torch.sub() 函数用于两个 tensor 间的相减
input: 输入的被减数,格式为tensor格式
other:输入的减数
alpha:与上面other参数搭配使用,用来与other相乘,当使用torch.sub()函数时不指定alpha的值时,alpha默认为1
out: 指定torch.sub()输出值被赋给的变量,可不指定