Skip to content

Commit 63f5562

Browse files
committed
ToString methods
Signed-off-by: Ryan Nett <[email protected]>
1 parent 665816c commit 63f5562

File tree

7 files changed

+54
-0
lines changed

7 files changed

+54
-0
lines changed

ndarray/src/main/java/org/tensorflow/ndarray/index/All.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,4 +49,9 @@ public boolean beginMask() {
4949
public boolean endMask() {
5050
return true;
5151
}
52+
53+
@Override
54+
public String toString() {
55+
return "All()";
56+
}
5257
}

ndarray/src/main/java/org/tensorflow/ndarray/index/At.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package org.tensorflow.ndarray.index;
1818

19+
import java.util.StringJoiner;
1920
import org.tensorflow.ndarray.impl.dimension.Dimension;
2021

2122
final class At implements TensorIndex {
@@ -68,4 +69,12 @@ public long end() {
6869
public boolean shrinkAxisMask() {
6970
return !keepDim;
7071
}
72+
73+
@Override
74+
public String toString() {
75+
return new StringJoiner(", ", At.class.getSimpleName() + "(", ")")
76+
.add("coord=" + coord)
77+
.add("keepDim=" + keepDim)
78+
.toString();
79+
}
7180
}

ndarray/src/main/java/org/tensorflow/ndarray/index/Ellipsis.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package org.tensorflow.ndarray.index;
1818

19+
import java.util.StringJoiner;
1920
import org.tensorflow.ndarray.impl.dimension.Dimension;
2021

2122
final class Ellipsis implements TensorIndex{
@@ -45,4 +46,9 @@ public boolean isEllipsis() {
4546
public boolean ellipsisMask() {
4647
return true;
4748
}
49+
50+
@Override
51+
public String toString() {
52+
return "Ellipsis()";
53+
}
4854
}

ndarray/src/main/java/org/tensorflow/ndarray/index/Hyperslab.java

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
*/
1616
package org.tensorflow.ndarray.index;
1717

18+
import java.util.StringJoiner;
1819
import org.tensorflow.ndarray.impl.dimension.Dimension;
1920

2021
/**
@@ -71,4 +72,14 @@ public boolean isPoint() {
7172
private final long stride;
7273
private final long count;
7374
private final long block;
75+
76+
@Override
77+
public String toString() {
78+
return new StringJoiner(", ", Hyperslab.class.getSimpleName() + "(", ")")
79+
.add("start=" + start)
80+
.add("stride=" + stride)
81+
.add("count=" + count)
82+
.add("block=" + block)
83+
.toString();
84+
}
7485
}

ndarray/src/main/java/org/tensorflow/ndarray/index/NewAxis.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,4 +50,9 @@ public boolean isNewAxis() {
5050
public boolean newAxisMask() {
5151
return true;
5252
}
53+
54+
@Override
55+
public String toString() {
56+
return "NewAxis()";
57+
}
5358
}

ndarray/src/main/java/org/tensorflow/ndarray/index/Sequence.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package org.tensorflow.ndarray.index;
1818

19+
import java.util.StringJoiner;
1920
import org.tensorflow.ndarray.NdArray;
2021
import org.tensorflow.ndarray.impl.dimension.Dimension;
2122

@@ -36,4 +37,11 @@ public long mapCoordinate(long coordinate, Dimension dim) {
3637
}
3738

3839
private final NdArray<? extends Number> coords;
40+
41+
@Override
42+
public String toString() {
43+
return new StringJoiner(", ", Sequence.class.getSimpleName() + "(", ")")
44+
.add("coords=" + coords)
45+
.toString();
46+
}
3947
}

ndarray/src/main/java/org/tensorflow/ndarray/index/Slice.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
*/
1717
package org.tensorflow.ndarray.index;
1818

19+
import java.util.StringJoiner;
1920
import org.tensorflow.ndarray.impl.dimension.Dimension;
2021

2122
final class Slice implements TensorIndex {
@@ -98,4 +99,13 @@ public boolean beginMask() {
9899
public boolean endMask() {
99100
return end == null;
100101
}
102+
103+
@Override
104+
public String toString() {
105+
return new StringJoiner(", ", Slice.class.getSimpleName() + "(", ")")
106+
.add("start=" + start)
107+
.add("end=" + end)
108+
.add("stride=" + stride)
109+
.toString();
110+
}
101111
}

0 commit comments

Comments
 (0)