class tensorflow::TensorShapeUtils
Static helper routines for TensorShape
. Includes a few common predicates on a tensor shape.
Member Details
static bool tensorflow::TensorShapeUtils::IsScalar(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsVector(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsVectorOrHigher(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsMatrix(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsSquareMatrix(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsMatrixOrHigher(const TensorShape &shape)
static Status tensorflow::TensorShapeUtils::MakeShape(const int32 *dims, int64 n, TensorShape *out)
Returns a TensorShape
whose dimensions are dims[0]
, dims[1]
, ..., dims[n-1]
.