DeepJava/tensorgraph.java at master · dido1998/DeepJava · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
packagetensordef;
importbasicops.*;
//import tensordef.*;
importjava.util.*;


publicclasstensorgraph
{
ArrayList<backpropagationstructure<?>> oplist;
publictensorgraph()
	{
oplist=newArrayList<backpropagationstructure<?>>();
	}
public <Textendssuperopdef>voidaddtolist(backpropagationstructure<T> ob)
	{
oplist.add(ob);
	}

public <Textendssuperopdef> voidremovefromlist(backpropagationstructure<T> ob)
	{
oplist.remove(ob);
	}
publicvoidbackward()
	{
backpropagationstructure<?> temp=oplist.get(oplist.size()-1);
if(temp.ans!=null)
 {
for(inti=0;i<temp.ans.dim1;i++)
 {
for(intj=0;j<temp.ans.dim2;j++)
 {
temp.ans.arr[i][j].grad=1;
 }
 }
 }else
 {
for(inti=0;i<temp.ans1.dim1;i++)
 {
for(intj=0;j<temp.ans1.dim2;j++)
 {
for(intk=0;k<temp.ans1.dim3;k++)
 {
temp.ans1.arr[i][j][k].grad=1;
 }
 }
 }
 }


while(!oplist.isEmpty())
 {
backpropagationstructure<?> curobjectref=oplist.get(oplist.size()-1);
//System.out.println(curobjectref);
if(curobjectref.ans!=null)
curobjectref.ob.backward(curobjectref.ans);
else
 {
//System.out.println(curobjectref.ob);
curobjectref.ob.backwardconv(curobjectref.ans1);
 }
 }
	}
}