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].

static Status tensorflow::TensorShapeUtils::MakeShape(const int64 *dims, int64 n, TensorShape *out)

static Status tensorflow::TensorShapeUtils::MakeShape(gtl::ArraySlice< int32 > shape, TensorShape *out)

static Status tensorflow::TensorShapeUtils::MakeShape(gtl::ArraySlice< int64 > shape, TensorShape *out)

string tensorflow::TensorShapeUtils::ShapeListString(const gtl::ArraySlice< TensorShape > &shapes)

bool tensorflow::TensorShapeUtils::StartsWith(const TensorShape &shape0, const TensorShape &shape1)