Class tensorflow::TensorShapeUtils

优质

小牛编辑

127浏览

2023-12-01

Static helper routines for TensorShape. Includes a few common predicates on a tensor shape.

Member Summary

static bool tensorflow::TensorShapeUtils::IsScalar(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsVector(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsLegacyScalar(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsLegacyVector(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsVectorOrHigher(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsMatrix(const TensorShape &shape)
static bool tensorflow::TensorShapeUtils::IsMatrixOrHigher(const TensorShape &shape)
static TensorShape tensorflow::TensorShapeUtils::MakeShape(const T *dims, int n)
- Returns a TensorShape whose dimensions are dims[0], dims[1], ..., dims[n-1].
static string tensorflow::TensorShapeUtils::ShapeListString(const gtl::ArraySlice< TensorShape > &shapes)
static bool tensorflow::TensorShapeUtils::StartsWith(const TensorShape &shape0, const TensorShape &shape1)

Returns a TensorShape whose dimensions are dims[0], dims[1], ..., dims[n-1].