torch (version 0.0.2)

torch_isinf: Isinf

Description

Isinf

Arguments

tensor

(Tensor) A tensor to check

TEST

Returns a new tensor with boolean elements representing if each element is +/-INF or not.

Examples

# NOT RUN {
if (torch_is_installed()) {

torch_isinf(torch_tensor(c(1, Inf, 2, -Inf, NaN)))
}
# }