org.nd4j.linalg.api.ops.Op to org.nd4j.linalg.api.ops.ScalarOp