sub.java

package basicops;
import java.util.*;

public class sub
{
	tensor num1;
	tensor num2;
	public sub(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.grad-=backflow.grad;
		
	}
}