Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::TensorGeometry Struct Reference

Public Member Functions

 TensorGeometry (IntArrayRef sizes)
 
 TensorGeometry (const Tensor &t)
 
bool is_contiguous () const
 
int64_t dim () const
 
int64_t size (int64_t dim) const
 
IntArrayRef sizes () const
 
int64_t stride (int64_t dim) const
 
IntArrayRef strides () const
 
int64_t storage_offset () const
 
int64_t numel () const
 
TensorGeometry transpose (int64_t dim0, int64_t dim1)
 

Data Fields

std::vector< int64_t > sizes_
 
std::vector< int64_t > strides_
 
int64_t storage_offset_
 
int64_t numel_
 

Detailed Description

Definition at line 8 of file TensorGeometry.h.


The documentation for this struct was generated from the following files: