PyTorch torch.heaviside 函数
Pytorch torch 参考手册torch.heaviside 是 PyTorch 中用于计算 Heaviside 阶跃函数的函数。当输入值大于 0 时返回 1，等于 0 时返回 0.5，小于 0 时返回 0。
函数定义
torch.heaviside(input, values)
使用示例
实例
import torch
# 创建输入张量
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# 指定当输入为 0 时的返回值
values = torch.tensor([0.5])
# 计算 Heaviside 阶跃函数
x = torch.heaviside(input, values)
print(x)
# 创建输入张量
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# 指定当输入为 0 时的返回值
values = torch.tensor([0.5])
# 计算 Heaviside 阶跃函数
x = torch.heaviside(input, values)
print(x)