Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members Class Reference
Inheritance diagram for

Public Member Functions

def __init__ (self, tensors)
def __getitem__ (self, index)
def __len__ (self)
- Public Member Functions inherited from
def __getitem__ (self, index)
def __len__ (self)
def __add__ (self, other)

Public Attributes


Detailed Description

Dataset wrapping tensors.

Each sample will be retrieved by indexing tensors along the first dimension.

    *tensors (Tensor): tensors that have the same size of the first dimension.

Definition at line 26 of file

The documentation for this class was generated from the following file: