DeepJava/tensorarray.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
64
65
66
67
68
69
70
71
72
73
74
package tensordef;
import java.util.*;
import java.util.Random;
import basicops.*;

import java.math.*;

public class tensorarray
{
	public tensor arr[][];

	public int dim1, dim2;
	public boolean trainable;
	public tensorarray(int dim1,int dim2,boolean trainable)
	{
		arr=new tensor[dim1][dim2];
		this.dim1=dim1;
		this.dim2=dim2;
		this.trainable=trainable;
		randomintialize();
	}
	public void randomintialize()
	{
		Random rand = new Random();
		for (int i=0;i<dim1;i++)
		{
			for (int j=0;j<dim2;j++)
			{
				if(rand.nextDouble()<0.5)
					arr[i][j]=new tensor(rand.nextDouble(),trainable);
				else
					arr[i][j]=new tensor(-rand.nextDouble(),trainable);
				//System.out.println(arr[i][j].data);
			}
		}
	}

	public void ones()
	{
		for (int i=0;i<dim1;i++)
		{
			for (int j=0;j<dim2;j++)
			{
				arr[i][j].data=1;
				//System.out.println(arr[i][j].data);
			}
		}
	}
	public void assign(double customdata[][])
	{
		for(int i=0;i<dim1;i++)
		{
			for(int j=0;j<dim2;j++)
			{
				arr[i][j].data=customdata[i][j];
			}
		}
	}
	public void assigntensorarray(tensorarray t)
	{
		for(int i=0;i<dim1;i++)
		{
			for(int j=0;j<dim2;j++)
			{
				arr[i][j].data=t.arr[i][j].data;
			}
		}
	}
	public void assign(tensor ip)
	{
		arr[0][0]=ip;

	}
}