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: