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::IsMatrixOrHigher(const TensorShape &shape)

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

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

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

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

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

results matching ""

    No results matching ""