Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

additional reverse signatures 

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2019-11-19 13:54:07 +03:00
parent bbd59a3537
commit ce2ef20f96
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/transforms/custom/Reverse.java
View File

@ -47,6 +47,29 @@ public class Reverse extends DynamicCustomOp {
this.inPlace = true;
}
/**
* This constructor allows to specify axis for Reverse operation
* @param x
* @param axis
*/
public Reverse(INDArray x, int... axis){
super(new INDArray[]{x}, new INDArray[0]);
this.inPlace = false;
addIArgument(axis);
}
/**
* This constructor allows to specify axis for Reverse operation
* @param x
* @param axis
*/
public Reverse(INDArray x, INDArray z, int... axis){
super(new INDArray[]{x}, new INDArray[] {z});
this.inPlace = false;
addIArgument(axis);
}
/**
* Reverses whole array for compatibility with OldReverse.
*