From a354f6c7f4e90df4e6e81e8ab72fa6ad3ee8fc12 Mon Sep 17 00:00:00 2001
From: Divya Sivakumaran <57161682+divyas248@users.noreply.github.com>
Date: Wed, 16 Nov 2022 16:55:23 -0500
Subject: [PATCH] Update nn.rst

The channel number for 'in' is currently set as c which is incorrect because c is referring to the output channel number. Instead this needs to be set as t (which is the variable that iterates over the input channel numbers).
in[n,c,i+h,j+w] should be changed to in[n,t,i+h,j+w]
---
 docs/source/objax/nn.rst | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/objax/nn.rst b/docs/source/objax/nn.rst
index 5431426..740f828 100644
--- a/docs/source/objax/nn.rst
+++ b/docs/source/objax/nn.rst
@@ -76,7 +76,7 @@ objax.nn
     with kernel weight :math:`(k,k,C_{in},C_{out})` and bias :math:`(C_{out})` as follows:
 
     .. math::
-      \mathrm{out}[n,c,h,w] = \mathrm{b}[c] + \sum_{t=0}^{C_{in}-1}\sum_{i=0}^{k-1}\sum_{j=0}^{k-1} \mathrm{in}[n,c,i+h,j+w] \times \mathrm{w}[i,j,t,c]
+      \mathrm{out}[n,c,h,w] = \mathrm{b}[c] + \sum_{t=0}^{C_{in}-1}\sum_{i=0}^{k-1}\sum_{j=0}^{k-1} \mathrm{in}[n,t,i+h,j+w] \times \mathrm{w}[i,j,t,c]
 
     where :math:`H_{out}=H-k+1`, :math:`W_{out}=W-k+1`.
     Note that the implementation follows the definition of