Add accessors for divider/dividend on divide vertex
@@ -11,6 +11,9 @@ import static io.improbable.keanu.tensor.TensorShapeValidation.checkHasSingleNon
public class DivisionVertex extends DoubleBinaryOpVertex {
+ private final DoubleVertex dividend;
+ private final DoubleVertex divsor;
+
/**
* Divides one vertex by another
*
@@ -19,6 +22,16 @@ public class DivisionVertex extends DoubleBinaryOpVertex {
*/
public DivisionVertex(DoubleVertex a, DoubleVertex b) {
super(checkHasSingleNonScalarShapeOrAllScalar(a.getShape(), b.getShape()), a, b);
+ this.dividend = a;
+ this.divsor = b;
+ }
+
+ public DoubleVertex getDividend(){
+ return dividend;
+ }
+
+ public DoubleVertex getDivsor(){
+ return divsor;
}
@Override
