eq¶
dragon.vm.torch.
eq
(
input,
other,
out=None
)[source]¶Compute the element-wise equal comparison.
\[\text{out} = (\text{input} == \text{other}) \]- Parameters:
- input (dragon.vm.torch.Tensor) – The input tensor.
- other (Union[dragon.vm.torch.Tensor, number]) – The tensor to compare.
- out (dragon.vm.torch.Tensor, optional) – The output tensor.
- Returns:
dragon.vm.torch.Tensor – The output tensor.