PyTorch .bitwise_or()

In PyTorch, the .bitwise_or() function computes the element-wise bitwise OR operation between two input tensors. For each pair of bits in the binary representation of the input elements, it returns 1 if at least one of the bits is 1, and 0 otherwise. This function supports integer tensors (signed or unsigned) and boolean tensors (where it performs a bitwise OR that behaves like a logical OR).

Syntax

torch.bitwise_or(input, other, *, out=None)

Parameters:

• input: The first input tensor. Must be of integer or boolean type.
• other: The second input tensor. Must be broadcastable with input and of the same type.
• out (Optional): A tensor for storing the output result.

Return value:

The .bitwise_or() function returns a new tensor containing the result of applying the bitwise OR operation to each pair of elements in the input tensors.

Example: Using torch.bitwise_or() with Integer and Boolean Tensors

The following example illustrates the usage of the .bitwise_or() function in PyTorch:

import torch

# Integer tensor example
tensor1 = torch.tensor([5, 3], dtype=torch.int32)  # Binary: 0101, 0011
tensor2 = torch.tensor([3, 7], dtype=torch.int32)  # Binary: 0011, 0111
result_int = torch.bitwise_or(tensor1, tensor2)

print(result_int)

# Boolean tensor example
tensor_bool1 = torch.tensor([True, False])
tensor_bool2 = torch.tensor([False, False])
result_bool = torch.bitwise_or(tensor_bool1, tensor_bool2)

print(result_bool)

Copy to clipboard
Copy to clipboard

The above code produces the following output:

tensor([7, 7], dtype=torch.int32)
tensor([ True, False])

Copy to clipboard
Copy to clipboard

In this example:

• 5 | 3 (binary 0101 | 0011) results in 0111 (decimal 7).
• 3 | 7 (binary 0011 | 0111) results in 0111 (decimal 7).
• True | False evaluates to True.
• False | False evaluates to False.