- Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconvexample.java
More file actions
Latest commit
23 lines (21 loc) · 607 Bytes
/
Copy pathconvexample.java
File metadata and controls
23 lines (21 loc) · 607 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
importtensordef.tensorgraph;
importtensordef.tensorarray;
importtensordef.add_tensorarray;
importtensordef.mul_tensorarray;
importtensordef.dot;
importtensordef.sigmoid;
importtensordef.tensorarray3d;
importtensordef.convolution;
publicclassconvexample
{
publicstaticvoidmain(Stringargs[])
{
tensorgraphmod=newtensorgraph();
tensorarray3dt1=newtensorarray3d(25,25,3,true);
System.out.println(t1.arr[0][0][0].grad);
convolutionconv=newconvolution(t1,3,1,mod,"SAME");
tensorarray3deval1=conv.forwardconv();
mod.backward();
System.out.println(t1.arr[0][0][0].grad);
}
}