首页 > 编程知识 正文

pytorch 基础,pytorch框架介绍

时间:2023-05-05 11:09:45 阅读:260401 作者:3444

Hello,大家好!今天为大家讲解以下pytorch的sign函数。

torch.sign(input, out=None) → Tensor
该函数的作用就是输出input通过sign函数后的活泼的冬瓜,其中sign函数就是符号函数,如下图所示。

例子:

>>> a = torch.tensor([0.7, -1.2, 0., 2.3])>>> atensor([ 0.7000, -1.2000, 0.0000, 2.3000])>>> torch.sign(a)tensor([ 1., -1., 0., 1.])

 

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。