PyTorch torch.heaviside 関数
Pytorch torch リファレンスマニュアルtorch.heavisidePyTorch で 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)
その他の拡張