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

Public Member Functions

 BatchTensor (at::Tensor data, at::Tensor mask, at::Tensor dims)
 BatchTensor (const at::Tensor &data, int64_t batch_size)
 BatchTensor (const std::vector< at::Tensor > &datalist, at::Tensor dims)
const char * toString () const
at::IntArrayRef sizes () const
int64_t dim () const
std::vector< at::Tensorexamples ()
at::Tensor get_data ()
at::Tensor get_mask ()
at::Tensor get_dims ()

Data Fields

at::Tensor data
at::Tensor mask
at::Tensor dims

Detailed Description

Definition at line 10 of file BatchTensor.h.

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