Struct tch::vision::dataset::Dataset[−][src]

pub struct Dataset {
    pub train_images: Tensor,
    pub train_labels: Tensor,
    pub test_images: Tensor,
    pub test_labels: Tensor,
    pub labels: i64,
}

Fields

train_images: Tensortrain_labels: Tensortest_images: Tensortest_labels: Tensorlabels: i64

Implementations

impl Dataset[src]

pub fn train_iter(&self, batch_size: i64) -> Iter2ⓘ

Notable traits for Iter2

impl Iterator for Iter2 type Item = (Tensor, Tensor);
[src]

pub fn test_iter(&self, batch_size: i64) -> Iter2ⓘ

Notable traits for Iter2

impl Iterator for Iter2 type Item = (Tensor, Tensor);
[src]

Trait Implementations

impl Debug for Dataset[src]

Auto Trait Implementations

impl RefUnwindSafe for Dataset

impl Send for Dataset

impl !Sync for Dataset

impl Unpin for Dataset

impl UnwindSafe for Dataset

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized, [src]

impl<T> Borrow<T> for T where
    T: ?Sized, [src]

impl<T> BorrowMut<T> for T where
    T: ?Sized, [src]

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, [src]

impl<T, U> TryFrom<U> for T where
    U: Into<T>, [src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, [src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

impl<V, T> VZip<V> for T where
    V: MultiLane<T>,