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::Tensor >	examples ()

at::Tensor	get_data ()

at::Tensor	get_mask ()

at::Tensor	get_dims ()

Detailed Description

Definition at line 10 of file BatchTensor.h.

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

Public Member Functions