public BaseTransformOp(INDArray x, INDArray y, INDArray z, long n) { super(x, y, z, n); if (y != null) LinAlgExceptions.assertSameLength(x, y, z); else LinAlgExceptions.assertSameLength(x, z); }
@Override public INDArray axpy(Number da, INDArray dx, INDArray dy) { // if(!dx.isVector()) // throw new IllegalArgumentException("Unable to use axpy on a non vector"); LinAlgExceptions.assertSameLength(dx, dy); level1().axpy(dx.length(), da.doubleValue(), dx, dy); return dy; }
/** * Assign all of the elements in the given * ndarray to this ndarray * * @param arr the elements to assign * @return this */ @Override public IComplexNDArray assign(IComplexNDArray arr) { if (!arr.isScalar()) LinAlgExceptions.assertSameLength(this, arr); IComplexNDArray linear = linearView(); IComplexNDArray otherLinear = arr.linearView(); for (int i = 0; i < linear.length(); i++) { linear.putScalar(i, otherLinear.getComplex(i)); } return this; }
public BaseTransformOp(INDArray x, INDArray y, INDArray z, long n) { super(x, y, z, n); if (y != null) LinAlgExceptions.assertSameLength(x, y); LinAlgExceptions.assertSameLength(x, z); }
public BaseTransformOp(INDArray x, INDArray z) { super(x, z); LinAlgExceptions.assertSameLength(x, z); LinAlgExceptions.assertSameShape(x, z); }
@Override public INDArray axpy(Number da, INDArray dx, INDArray dy) { // if(!dx.isVector()) // throw new IllegalArgumentException("Unable to use axpy on a non vector"); LinAlgExceptions.assertSameLength(dx, dy); level1().axpy(dx.length(), da.doubleValue(), dx, dy); return dy; }
/** * Assign all of the elements in the given * ndarray to this ndarray * * @param arr the elements to assign * @return this */ @Override public IComplexNDArray assign(IComplexNDArray arr) { if (!arr.isScalar()) LinAlgExceptions.assertSameLength(this, arr); IComplexNDArray linear = linearView(); IComplexNDArray otherLinear = arr.linearView(); for (int i = 0; i < linear.length(); i++) { linear.putScalar(i, otherLinear.getComplex(i)); } return this; }