public class TensorShape
extends java.lang.Object
Constructor and Description |
---|
TensorShape(int[] shape) |
Modifier and Type | Method and Description |
---|---|
boolean |
equals(java.lang.Object o) |
static int |
getFlatIndex(int[] shape,
int[] stride,
int... index) |
static long |
getLength(int[] shape) |
static int[] |
getRowFirstStride(int[] shape) |
int[] |
getShape() |
int |
hashCode() |
boolean |
isScalar() |
static boolean |
isScalar(int[] shape) |
public int[] getShape()
public boolean isScalar()
public boolean equals(java.lang.Object o)
equals
in class java.lang.Object
public int hashCode()
hashCode
in class java.lang.Object
public static long getLength(int[] shape)
shape
- for finding length ofpublic static int[] getRowFirstStride(int[] shape)
shape
- shape to find stride forpublic static int getFlatIndex(int[] shape, int[] stride, int... index)
public static boolean isScalar(int[] shape)