6.1.10 Class Tensor Shape Utils
优质
小牛编辑
126浏览
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
TensorShapewhose dimensions aredims[0],dims[1], ...,dims[n-1].
- Returns a
static string tensorflow::TensorShapeUtils::ShapeListString(const gtl::ArraySlice< TensorShape > &shapes)static bool tensorflow::TensorShapeUtils::StartsWith(const TensorShape &shape0, const TensorShape &shape1)
Member Details
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].