torch (version 0.0.2)

torch_where: Where

Description

Where

Arguments

condition

(BoolTensor) When True (nonzero), yield x, otherwise yield y

x

(Tensor) values selected at indices where condition is True

y

(Tensor) values selected at indices where condition is False

where(condition, x, y) -> Tensor

Return a tensor of elements selected from either x or y, depending on condition.

The operation is defined as:

$$ \mbox{out}_i = \left\{ \begin{array}{ll} \mbox{x}_i & \mbox{if } \mbox{condition}_i \\ \mbox{y}_i & \mbox{otherwise} \\ \end{array} \right. $$

where(condition) -> tuple of LongTensor

torch_where(condition) is identical to torch_nonzero(condition, as_tuple=True).

Examples

# NOT RUN {
if (torch_is_installed()) {

# }
# NOT RUN {
x = torch_randn(c(3, 2))
y = torch_ones(c(3, 2))
x
torch_where(x > 0, x, y)
# }
# NOT RUN {


}
# }