Tensor Multiplication (Element-wise) with PyTorch and CUDA

import torch

# Check if CUDA is available and set the device
if torch.cuda.is_available():
    device = torch.device("cuda")
    print(f"Using CUDA device: {torch.cuda.get_device_name(0)}")
else:
    device = torch.device("cpu")
    print("CUDA not available, using CPU")

# Define two tensors of the same shape
tensor_a = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32).to(device)
tensor_b = torch.tensor([[7, 8, 9], [10, 11, 12]], dtype=torch.float32).to(device)

# Perform element-wise tensor multiplication
tensor_product = torch.mul(tensor_a, tensor_b)
# Alternatively, you can use the '*' operator:
# tensor_product = tensor_a * tensor_b

# Print the result
print("Tensor A:\n", tensor_a)
print("Tensor B:\n", tensor_b)
print("Element-wise Product of Tensor A and Tensor B:\n", tensor_product.cpu().numpy())