class tensorflow::TensorShape
Member Details
uint8 tensorflow::TensorShape::buf[16][16]
Rep64* tensorflow::TensorShape::unused_aligner
tensorflow::TensorShape::TensorShape(gtl::ArraySlice< int64 > dim_sizes)
Construct a TensorShape
from the provided sizes. REQUIRES: dim_sizes[i] >= 0
tensorflow::TensorShape::TensorShape(std::initializer_list< int64 > dim_sizes)
tensorflow::TensorShape::TensorShape(const TensorShapeProto &proto)
REQUIRES: IsValid(proto)
tensorflow::TensorShape::TensorShape()
Create a tensor shape with no dimensions and one element, which you can then call AddDim()
on.
tensorflow::TensorShape::~TensorShape()
tensorflow::TensorShape::TensorShape(const TensorShape &b)
Copy the specified shape.
void tensorflow::TensorShape::operator=(const TensorShape &b)
void tensorflow::TensorShape::Clear()
Clear a tensor shape.
void tensorflow::TensorShape::AddDim(int64 size)
Add a dimension to the end ("inner-most"). REQUIRES: size >= 0
void tensorflow::TensorShape::AppendShape(const TensorShape &shape)
Appends all the dimensions from shape
.
void tensorflow::TensorShape::InsertDim(int d, int64 size)
Insert a dimension somewhere in the TensorShape
. REQUIRES: 0 <= d <= dims()
REQUIRES: size >= 0
void tensorflow::TensorShape::set_dim(int d, int64 size)
Modifies the size of the dimension d
to be size
REQUIRES: 0 <= d < dims()
REQUIRES: size >= 0
void tensorflow::TensorShape::RemoveDim(int d)
Removes dimension d
from the TensorShape
. REQUIRES: 0 <= d < dims()
int tensorflow::TensorShape::dims() const
Return the number of dimensions in the tensor.
int64 tensorflow::TensorShape::dim_size(int d) const
Returns the number of elements in dimension d
. REQUIRES: 0 <= d < dims()
gtl::InlinedVector< int64, 4 > tensorflow::TensorShape::dim_sizes() const
Returns sizes of all dimensions.
int64 tensorflow::TensorShape::num_elements() const
Returns the number of elements in the tensor.
We use int64
and not size_t
to be compatible with Eigen::Tensor
which uses ptrdiff_t
.
bool tensorflow::TensorShape::IsSameSize(const TensorShape &b) const
Returns true if *this
and b
have the same sizes. Ignores dimension names.
bool tensorflow::TensorShape::operator==(const TensorShape &b) const
bool tensorflow::TensorShape::operator!=(const TensorShape &b) const
void tensorflow::TensorShape::AsProto(TensorShapeProto *proto) const
Fill *proto
from *this
.
Eigen::DSizes< Eigen::DenseIndex, NDIMS > tensorflow::TensorShape::AsEigenDSizes() const
Fill *dsizes
from *this
.
Eigen::DSizes< Eigen::DenseIndex, NDIMS > tensorflow::TensorShape::AsEigenDSizesWithPadding() const
Same as AsEigenDSizes()
but allows for NDIMS > dims()
in which case we pad the rest of the sizes with 1.
TensorShapeIter tensorflow::TensorShape::begin() const
For iterating through the dimensions.
TensorShapeIter tensorflow::TensorShape::end() const
string tensorflow::TensorShape::DebugString() const
For error messages.
void tensorflow::TensorShape::DumpRep() const
bool tensorflow::TensorShape::IsValid(const TensorShapeProto &proto)
Returns true
iff proto
is a valid tensor shape.
Status tensorflow::TensorShape::IsValidShape(const TensorShapeProto &proto)
Returns OK
iff proto
is a valid tensor shape, and a descriptive error status otherwise.
static constexpr int tensorflow::TensorShape::MaxDimensions()
string tensorflow::TensorShape::DebugString(const TensorShapeProto &proto)
Same as TensorShape(proto). DebugString()
but doesn't crash for invalid protos.