torch.copysign¶
- torch.copysign(input, other, *, out=None) Tensor ¶
按元素方式创建一个新的浮点张量,其幅度为
input
的幅度,符号为other
的符号。支持 广播到公共形状,以及整数和浮点输入。
- 参数
- 关键字参数
out (Tensor, 可选) – 输出张量。
示例
>>> a = torch.randn(5) >>> a tensor([-1.2557, -0.0026, -0.5387, 0.4740, -0.9244]) >>> torch.copysign(a, 1) tensor([1.2557, 0.0026, 0.5387, 0.4740, 0.9244]) >>> a = torch.randn(4, 4) >>> a tensor([[ 0.7079, 0.2778, -1.0249, 0.5719], [-0.0059, -0.2600, -0.4475, -1.3948], [ 0.3667, -0.9567, -2.5757, -0.1751], [ 0.2046, -0.0742, 0.2998, -0.1054]]) >>> b = torch.randn(4) tensor([ 0.2373, 0.3120, 0.3190, -1.1128]) >>> torch.copysign(a, b) tensor([[ 0.7079, 0.2778, 1.0249, -0.5719], [ 0.0059, 0.2600, 0.4475, -1.3948], [ 0.3667, 0.9567, 2.5757, -0.1751], [ 0.2046, 0.0742, 0.2998, -0.1054]]) >>> a = torch.tensor([1.]) >>> b = torch.tensor([-0.]) >>> torch.copysign(a, b) tensor([-1.])
注意
copysign 处理带符号的零。如果 other 参数具有负零 (-0),则相应的输出值将为负。