mul.java

package basicops;
import java.util.*;

public class mul
{
	tensor num1;
	tensor num2;
	public mul(tensor a,tensor b)
	{
		num1=a;
		num2=b;
	}
	public tensor forward()
	{
		tensor evaltensor=new tensor(num1.data*num2.data,false);
		return evaltensor;
	}
	public void backward(tensor backflow)
	{
		num1.grad+=backflow.grad*num2.data;
		num2.grad+=backflow.grad*num1.data;
	}
}