Pytorch使用符号函数

最新推荐文章于 2023-04-15 18:28:06 发布

飞机火车巴雷特

最新推荐文章于 2023-04-15 18:28:06 发布

阅读量1.1k

点赞数 1

分类专栏： Python相关

本文链接：https://blog.csdn.net/qq_36158230/article/details/121635437

版权

pytorch

Python相关专栏收录该内容

61 篇文章 1 订阅

订阅专栏

一、定义

符号函数的定义如下

$sign(x)=\left\{\begin{matrix}1, x > 0 \\ 0, x = 0 \\ -1, x < 0 \end{matrix}\right.$

二、代码

import torch

# set print option preference
torch.set_printoptions(linewidth=1024, sci_mode=False)

# 1D (vector)
a = torch.tensor([-99.0, -1.0, -0.5, 0.0, 0.0, 0.5, 1.5, 2.5])

# 2D (tensor)
b = torch.tensor([[-99.0, -1.0, -0.5],
                  [0.0, 0.0, 0.0],
                  [1.0, 3.0, 10.5],
                  [-0.1, 0.0, 2.0]])
# use sign function
after_a = torch.sign(a)
after_b = torch.sign(b)

# print
print(a)
print(after_a)
print(b)
print(after_b)

输出

tensor([-99.0000, -1.0000, -0.5000, 0.0000, 0.0000, 0.5000, 1.5000, 2.5000])
tensor([-1., -1., -1., 0., 0., 1., 1., 1.])

tensor([[-99.0000, -1.0000, -0.5000],
[ 0.0000, 0.0000, 0.0000],
[ 1.0000, 3.0000, 10.5000],
[ -0.1000, 0.0000, 2.0000]])
tensor([[-1., -1., -1.],
[ 0., 0., 0.],
[ 1., 1., 1.],
[-1., 0., 1.]])