From 27f1d666c187213f1b359ec83043bc8d0fed051c Mon Sep 17 00:00:00 2001
From: raviguptaamd <ravi.gupta@amd.com>
Date: Mon, 28 Apr 2025 17:55:29 -0500
Subject: [PATCH] Adding nn_Conv2d examples

---
 rg_examples/nn_Conv2d.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 rg_examples/nn_Conv2d.py

diff --git a/rg_examples/nn_Conv2d.py b/rg_examples/nn_Conv2d.py
new file mode 100644
index 00000000000000..2175df10ba48bd
--- /dev/null
+++ b/rg_examples/nn_Conv2d.py
@@ -0,0 +1,16 @@
+import torch
+import torch.nn as nn
+
+
+# Create a random input tensor
+in_tensor = torch.randn(1, 3, 448, 448).to('cuda')
+
+# Create a simple convolutional layer
+conv_2d = nn.Conv2d(in_channels=3, out_channels=32, kernel_size=3, padding=1).to('cuda')
+
+# Perform convolution
+output = conv_2d(in_tensor)
+
+print(f"Input shape: {in_tensor.shape}")
+print(f"Output shape: {output.shape}")
+print("Convolution completed successfully")