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

Update Nd4jTestsC.java

Browse Source
This commit is contained in:
agibsonccc 2021-03-23 17:38:45 +09:00
parent bcce1f8db4
commit ac38c704cd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/Nd4jTestsC.java
View File

@ -2413,7 +2413,7 @@ public class Nd4jTestsC extends BaseNd4jTestWithBackends {
@ParameterizedTest
@MethodSource("org.nd4j.linalg.BaseNd4jTestWithBackends#configs")
public void testPutAtIntervalIndexWithStride(Nd4jBackend backend) {
INDArray n1 = Nd4j.create(3, 3).assign(0.0).castTo(DataType.DOUBLE));
INDArray n1 = Nd4j.create(3, 3).assign(0.0).castTo(DataType.DOUBLE);
INDArrayIndex[] indices = {NDArrayIndex.interval(0, 2, 3), NDArrayIndex.all()};
n1.put(indices, 1);
INDArray expected = Nd4j.create(new double[][] {{1d, 1d, 1d}, {0d, 0d, 0d}, {1d, 1d, 1d}});