Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

torch_heaviside

Heaviside


Description

Heaviside

Usage

torch_heaviside(self, values)

Arguments

self

(Tensor) the input tensor.

values

(Tensor) The values to use where input is zero.

heaviside(input, values, *, out=None) -> Tensor

Computes the Heaviside step function for each element in input. The Heaviside step function is defined as:

\mbox{{heaviside}}(input, values) = \begin{array}{ll} 0, & \mbox{if input < 0}\\ values, & \mbox{if input == 0}\\ 1, & \mbox{if input > 0} \end{array}

Examples

if (torch_is_installed()) {

input <- torch_tensor(c(-1.5, 0, 2.0))
values <- torch_tensor(c(0.5))
torch_heaviside(input, values)
values <- torch_tensor(c(1.2, -2.0, 3.5))
torch_heaviside(input, values)
}

torch

Tensors and Neural Networks with 'GPU' Acceleration

v0.3.0
MIT + file LICENSE
Authors
Daniel Falbel [aut, cre, cph], Javier Luraschi [aut], Dmitriy Selivanov [ctb], Athos Damiani [ctb], Christophe Regouby [ctb], Krzysztof Joachimiak [ctb], RStudio [cph]
Initial release

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.