torch.square(input,*,out = None)→张量
返回带有输入元素平方的新张量。
参数
输入(张量)–输入张量。
关键字参数
out(张量,可选)–输出张量。
例子:
>>> a = torch.randn(4)
>>> a
tensor([-2.0755, 1.0226, 0.0831, 0.4806])
>>> torch.square(a)
tensor([ 4.3077, 1.0457, 0.0069, 0.2310])