Trait ai_dataloader::collate::Collate

pub trait Collate<T> {
    type Output;

    // Required method
    fn collate(&self, batch: Vec<T>) -> Self::Output;
}

Any collate gather samples from one batch together.

A DefaultCollate struct is provided which will cover most of the use cases.

This trait is used instead of Fn because we cannot currently impl Fn* on struct on stable rust.

Required Associated Types

type Output

The type of the collate function’s output

Required Methods

fn collate(&self, batch: Vec<T>) -> Self::Output

Take a batch of samples and collate them

Implementors

impl Collate<bool> for DefaultCollate

type Output = ArrayBase<OwnedRepr<bool>, Dim<[usize; 1]>>

impl Collate<bool> for TorchCollate

Available on crate feature tch only.

type Output = Tensor

impl Collate<char> for DefaultCollate

type Output = ArrayBase<OwnedRepr<char>, Dim<[usize; 1]>>

impl Collate<f32> for DefaultCollate

type Output = ArrayBase<OwnedRepr<f32>, Dim<[usize; 1]>>

impl Collate<f32> for TorchCollate

Available on crate feature tch only.

type Output = Tensor

impl Collate<f64> for DefaultCollate

type Output = ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>

impl Collate<f64> for TorchCollate

Available on crate feature tch only.

type Output = Tensor

impl Collate<i8> for DefaultCollate

type Output = ArrayBase<OwnedRepr<i8>, Dim<[usize; 1]>>

impl Collate<i8> for TorchCollate

Available on crate feature tch only.

type Output = Tensor

impl Collate<i16> for DefaultCollate

type Output = ArrayBase<OwnedRepr<i16>, Dim<[usize; 1]>>

impl Collate<i16> for TorchCollate

Available on crate feature tch only.

type Output = Tensor

impl Collate<i32> for DefaultCollate

type Output = ArrayBase<OwnedRepr<i32>, Dim<[usize; 1]>>

impl Collate<i32> for TorchCollate

Available on crate feature tch only.

type Output = Tensor

impl Collate<i64> for DefaultCollate

type Output = ArrayBase<OwnedRepr<i64>, Dim<[usize; 1]>>

impl Collate<i64> for TorchCollate

Available on crate feature tch only.

type Output = Tensor

impl Collate<i128> for DefaultCollate

type Output = ArrayBase<OwnedRepr<i128>, Dim<[usize; 1]>>

impl Collate<isize> for DefaultCollate

type Output = ArrayBase<OwnedRepr<isize>, Dim<[usize; 1]>>

impl Collate<u8> for DefaultCollate

NoOp for binairy, as pytorch default_collate function.

type Output = Vec<u8, Global>

impl Collate<u8> for TorchCollate

Available on crate feature tch only.

NoOp for binairy, as pytorch default_collate function.

type Output = Vec<u8, Global>

impl Collate<u16> for DefaultCollate

type Output = ArrayBase<OwnedRepr<u16>, Dim<[usize; 1]>>

impl Collate<u32> for DefaultCollate

type Output = ArrayBase<OwnedRepr<u32>, Dim<[usize; 1]>>

impl Collate<u64> for DefaultCollate

type Output = ArrayBase<OwnedRepr<u64>, Dim<[usize; 1]>>

impl Collate<u128> for DefaultCollate

type Output = ArrayBase<OwnedRepr<u128>, Dim<[usize; 1]>>

impl Collate<usize> for DefaultCollate

type Output = ArrayBase<OwnedRepr<usize>, Dim<[usize; 1]>>

impl Collate<CString> for DefaultCollate

type Output = Vec<CString, Global>

impl Collate<CString> for TorchCollate

Available on crate feature tch only.

type Output = Vec<CString, Global>

impl Collate<String> for DefaultCollate

type Output = Vec<String, Global>

impl Collate<String> for TorchCollate

Available on crate feature tch only.

type Output = Vec<String, Global>

impl Collate<NonZeroI8> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroI8>, Dim<[usize; 1]>>

impl Collate<NonZeroI8> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroI8>, Dim<[usize; 1]>>

impl Collate<NonZeroI16> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroI16>, Dim<[usize; 1]>>

impl Collate<NonZeroI16> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroI16>, Dim<[usize; 1]>>

impl Collate<NonZeroI32> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroI32>, Dim<[usize; 1]>>

impl Collate<NonZeroI32> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroI32>, Dim<[usize; 1]>>

impl Collate<NonZeroI64> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroI64>, Dim<[usize; 1]>>

impl Collate<NonZeroI64> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroI64>, Dim<[usize; 1]>>

impl Collate<NonZeroI128> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroI128>, Dim<[usize; 1]>>

impl Collate<NonZeroI128> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroI128>, Dim<[usize; 1]>>

impl Collate<NonZeroIsize> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroIsize>, Dim<[usize; 1]>>

impl Collate<NonZeroIsize> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroIsize>, Dim<[usize; 1]>>

impl Collate<NonZeroU8> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroU8>, Dim<[usize; 1]>>

impl Collate<NonZeroU8> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroU8>, Dim<[usize; 1]>>

impl Collate<NonZeroU16> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroU16>, Dim<[usize; 1]>>

impl Collate<NonZeroU16> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroU16>, Dim<[usize; 1]>>

impl Collate<NonZeroU32> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroU32>, Dim<[usize; 1]>>

impl Collate<NonZeroU32> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroU32>, Dim<[usize; 1]>>

impl Collate<NonZeroU64> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroU64>, Dim<[usize; 1]>>

impl Collate<NonZeroU64> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroU64>, Dim<[usize; 1]>>

impl Collate<NonZeroU128> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroU128>, Dim<[usize; 1]>>

impl Collate<NonZeroU128> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroU128>, Dim<[usize; 1]>>

impl Collate<NonZeroUsize> for DefaultCollate

type Output = ArrayBase<OwnedRepr<NonZeroUsize>, Dim<[usize; 1]>>

impl Collate<NonZeroUsize> for TorchCollate

Available on crate feature tch only.

type Output = ArrayBase<OwnedRepr<NonZeroUsize>, Dim<[usize; 1]>>

impl Collate<OsString> for DefaultCollate

type Output = Vec<OsString, Global>

impl Collate<OsString> for TorchCollate

Available on crate feature tch only.

type Output = Vec<OsString, Global>

impl<'a> Collate<&'a str> for DefaultCollate

type Output = Vec<&'a str, Global>

impl<'a> Collate<&'a str> for TorchCollate

Available on crate feature tch only.

type Output = Vec<&'a str, Global>

impl<'a> Collate<&'a CStr> for DefaultCollate

type Output = Vec<&'a CStr, Global>

impl<'a> Collate<&'a CStr> for TorchCollate

Available on crate feature tch only.

type Output = Vec<&'a CStr, Global>

impl<A> Collate<(A,)> for DefaultCollatewhere A: Clone, DefaultCollate: Collate<A>,

type Output = (<DefaultCollate as Collate<A>>::Output,)

impl<A> Collate<(A,)> for TorchCollatewhere A: Clone, TorchCollate: Collate<A>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output,)

impl<A, B> Collate<(A, B)> for DefaultCollatewhere A: Clone, B: Clone, DefaultCollate: Collate<A> + Collate<B>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output)

impl<A, B> Collate<(A, B)> for TorchCollatewhere A: Clone, B: Clone, TorchCollate: Collate<A> + Collate<B>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output)

impl<A, B, C> Collate<(A, B, C)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output)

impl<A, B, C> Collate<(A, B, C)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output)

impl<A, B, C, D> Collate<(A, B, C, D)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output)

impl<A, B, C, D> Collate<(A, B, C, D)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output)

impl<A, B, C, D, E> Collate<(A, B, C, D, E)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output)

impl<A, B, C, D, E> Collate<(A, B, C, D, E)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output)

impl<A, B, C, D, E, F> Collate<(A, B, C, D, E, F)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output, <DefaultCollate as Collate<F>>::Output)

impl<A, B, C, D, E, F> Collate<(A, B, C, D, E, F)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output, <TorchCollate as Collate<F>>::Output)

impl<A, B, C, D, E, F, G> Collate<(A, B, C, D, E, F, G)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output, <DefaultCollate as Collate<F>>::Output, <DefaultCollate as Collate<G>>::Output)

impl<A, B, C, D, E, F, G> Collate<(A, B, C, D, E, F, G)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output, <TorchCollate as Collate<F>>::Output, <TorchCollate as Collate<G>>::Output)

impl<A, B, C, D, E, F, G, H> Collate<(A, B, C, D, E, F, G, H)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output, <DefaultCollate as Collate<F>>::Output, <DefaultCollate as Collate<G>>::Output, <DefaultCollate as Collate<H>>::Output)

impl<A, B, C, D, E, F, G, H> Collate<(A, B, C, D, E, F, G, H)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output, <TorchCollate as Collate<F>>::Output, <TorchCollate as Collate<G>>::Output, <TorchCollate as Collate<H>>::Output)

impl<A, B, C, D, E, F, G, H, I> Collate<(A, B, C, D, E, F, G, H, I)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output, <DefaultCollate as Collate<F>>::Output, <DefaultCollate as Collate<G>>::Output, <DefaultCollate as Collate<H>>::Output, <DefaultCollate as Collate<I>>::Output)

impl<A, B, C, D, E, F, G, H, I> Collate<(A, B, C, D, E, F, G, H, I)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output, <TorchCollate as Collate<F>>::Output, <TorchCollate as Collate<G>>::Output, <TorchCollate as Collate<H>>::Output, <TorchCollate as Collate<I>>::Output)

impl<A, B, C, D, E, F, G, H, I, J> Collate<(A, B, C, D, E, F, G, H, I, J)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, J: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I> + Collate<J>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output, <DefaultCollate as Collate<F>>::Output, <DefaultCollate as Collate<G>>::Output, <DefaultCollate as Collate<H>>::Output, <DefaultCollate as Collate<I>>::Output, <DefaultCollate as Collate<J>>::Output)

impl<A, B, C, D, E, F, G, H, I, J> Collate<(A, B, C, D, E, F, G, H, I, J)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, J: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I> + Collate<J>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output, <TorchCollate as Collate<F>>::Output, <TorchCollate as Collate<G>>::Output, <TorchCollate as Collate<H>>::Output, <TorchCollate as Collate<I>>::Output, <TorchCollate as Collate<J>>::Output)

impl<A, B, C, D, E, F, G, H, I, J, K> Collate<(A, B, C, D, E, F, G, H, I, J, K)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, J: Clone, K: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I> + Collate<J> + Collate<K>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output, <DefaultCollate as Collate<F>>::Output, <DefaultCollate as Collate<G>>::Output, <DefaultCollate as Collate<H>>::Output, <DefaultCollate as Collate<I>>::Output, <DefaultCollate as Collate<J>>::Output, <DefaultCollate as Collate<K>>::Output)

impl<A, B, C, D, E, F, G, H, I, J, K> Collate<(A, B, C, D, E, F, G, H, I, J, K)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, J: Clone, K: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I> + Collate<J> + Collate<K>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output, <TorchCollate as Collate<F>>::Output, <TorchCollate as Collate<G>>::Output, <TorchCollate as Collate<H>>::Output, <TorchCollate as Collate<I>>::Output, <TorchCollate as Collate<J>>::Output, <TorchCollate as Collate<K>>::Output)

impl<A, B, C, D, E, F, G, H, I, J, K, L> Collate<(A, B, C, D, E, F, G, H, I, J, K, L)> for DefaultCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, J: Clone, K: Clone, L: Clone, DefaultCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I> + Collate<J> + Collate<K> + Collate<L>,

type Output = (<DefaultCollate as Collate<A>>::Output, <DefaultCollate as Collate<B>>::Output, <DefaultCollate as Collate<C>>::Output, <DefaultCollate as Collate<D>>::Output, <DefaultCollate as Collate<E>>::Output, <DefaultCollate as Collate<F>>::Output, <DefaultCollate as Collate<G>>::Output, <DefaultCollate as Collate<H>>::Output, <DefaultCollate as Collate<I>>::Output, <DefaultCollate as Collate<J>>::Output, <DefaultCollate as Collate<K>>::Output, <DefaultCollate as Collate<L>>::Output)

impl<A, B, C, D, E, F, G, H, I, J, K, L> Collate<(A, B, C, D, E, F, G, H, I, J, K, L)> for TorchCollatewhere A: Clone, B: Clone, C: Clone, D: Clone, E: Clone, F: Clone, G: Clone, H: Clone, I: Clone, J: Clone, K: Clone, L: Clone, TorchCollate: Collate<A> + Collate<B> + Collate<C> + Collate<D> + Collate<E> + Collate<F> + Collate<G> + Collate<H> + Collate<I> + Collate<J> + Collate<K> + Collate<L>,

Available on crate feature tch only.

type Output = (<TorchCollate as Collate<A>>::Output, <TorchCollate as Collate<B>>::Output, <TorchCollate as Collate<C>>::Output, <TorchCollate as Collate<D>>::Output, <TorchCollate as Collate<E>>::Output, <TorchCollate as Collate<F>>::Output, <TorchCollate as Collate<G>>::Output, <TorchCollate as Collate<H>>::Output, <TorchCollate as Collate<I>>::Output, <TorchCollate as Collate<J>>::Output, <TorchCollate as Collate<K>>::Output, <TorchCollate as Collate<L>>::Output)

impl<A, D> Collate<ArrayBase<OwnedRepr<A>, D>> for DefaultCollatewhere A: Clone, D: Dimension, D::Larger: RemoveAxis,

type Output = ArrayBase<OwnedRepr<A>, <D as Dimension>::Larger>

impl<A, D> Collate<ArrayBase<OwnedRepr<A>, D>> for TorchCollatewhere A: Clone + Element, D: Dimension, D::Larger: RemoveAxis,

Available on crate feature tch only.

type Output = Tensor

impl<K, V> Collate<BTreeMap<K, V, Global>> for DefaultCollatewhere K: Ord + Clone, V: Clone, Self: Collate<V>,

type Output = BTreeMap<K, <DefaultCollate as Collate<V>>::Output, Global>

impl<K, V> Collate<BTreeMap<K, V, Global>> for TorchCollatewhere K: Ord + Clone, V: Clone, Self: Collate<V>,

Available on crate feature tch only.

type Output = BTreeMap<K, <TorchCollate as Collate<V>>::Output, Global>

impl<K, V, H> Collate<HashMap<K, V, H>> for DefaultCollatewhere K: Eq + Hash + Clone, V: Clone, Self: Collate<V>, H: BuildHasher,

type Output = HashMap<K, <DefaultCollate as Collate<V>>::Output, RandomState>

impl<K, V, H> Collate<HashMap<K, V, H>> for TorchCollatewhere K: Eq + Hash + Clone, V: Clone, Self: Collate<V>, H: BuildHasher,

Available on crate feature tch only.

type Output = HashMap<K, <TorchCollate as Collate<V>>::Output, RandomState>

impl<T> Collate<&T> for DefaultCollatewhere T: Clone, Self: Collate<T>,

We think it makes no sense to but a bench of reference into a Tensor. That’s why if the dataset yield reference a we clone them them. It is useful for having a non-consuming Iterator over the Dataloader.

type Output = <DefaultCollate as Collate<T>>::Output

impl<T> Collate<&T> for TorchCollatewhere T: Clone, Self: Collate<T>,

Available on crate feature tch only.

We think it makes no sense to but a bench of reference into a Tensor. That’s why if the dataset yield reference and they we clone their value. It is useful for having a non-consuming Iterator over the Dataloader.

type Output = <TorchCollate as Collate<T>>::Output

impl<T> Collate<VecDeque<T, Global>> for DefaultCollatewhere Self: Collate<T>, T: Clone,

type Output = Vec<<DefaultCollate as Collate<T>>::Output, Global>

impl<T> Collate<VecDeque<T, Global>> for TorchCollatewhere Self: Collate<T>, T: Clone,

Available on crate feature tch only.

type Output = Vec<<TorchCollate as Collate<T>>::Output, Global>

impl<T> Collate<Vec<T, Global>> for DefaultCollatewhere Self: Collate<T>, T: Clone,

type Output = Vec<<DefaultCollate as Collate<T>>::Output, Global>

impl<T> Collate<Vec<T, Global>> for TorchCollatewhere Self: Collate<T>, T: Clone,

Available on crate feature tch only.

type Output = Vec<<TorchCollate as Collate<T>>::Output, Global>

impl<T> Collate<T> for NoOpCollate

type Output = Vec<T, Global>

impl<T, F, O> Collate<T> for Fwhere F: Fn(Vec<T>) -> O,

type Output = O

impl<T, const N: usize> Collate<[T; N]> for DefaultCollatewhere Self: Collate<T>, T: Clone,

type Output = Vec<<DefaultCollate as Collate<T>>::Output, Global>

impl<T, const N: usize> Collate<[T; N]> for TorchCollatewhere Self: Collate<T>, T: Clone,

Available on crate feature tch only.

type Output = Vec<<TorchCollate as Collate<T>>::Output, Global>