Trait 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 binary, as pytorch default_collate function.

type Output = Vec<u8>

impl Collate<u8> for TorchCollate

Available on crate feature tch only.

NoOp for binary, as pytorch default_collate function.

type Output = Vec<u8>

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>

impl Collate<CString> for TorchCollate

Available on crate feature tch only.

type Output = Vec<CString>

impl Collate<String> for DefaultCollate

type Output = Vec<String>

impl Collate<String> for TorchCollate

Available on crate feature tch only.

type Output = Vec<String>

impl Collate<NonZero<i8>> for DefaultCollate

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

impl Collate<NonZero<i8>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<i16>> for DefaultCollate

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

impl Collate<NonZero<i16>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<i32>> for DefaultCollate

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

impl Collate<NonZero<i32>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<i64>> for DefaultCollate

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

impl Collate<NonZero<i64>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<i128>> for DefaultCollate

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

impl Collate<NonZero<i128>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<isize>> for DefaultCollate

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

impl Collate<NonZero<isize>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<u8>> for DefaultCollate

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

impl Collate<NonZero<u8>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<u16>> for DefaultCollate

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

impl Collate<NonZero<u16>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<u32>> for DefaultCollate

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

impl Collate<NonZero<u32>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<u64>> for DefaultCollate

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

impl Collate<NonZero<u64>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<u128>> for DefaultCollate

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

impl Collate<NonZero<u128>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<NonZero<usize>> for DefaultCollate

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

impl Collate<NonZero<usize>> for TorchCollate

Available on crate feature tch only.

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

impl Collate<OsString> for DefaultCollate

type Output = Vec<OsString>

impl Collate<OsString> for TorchCollate

Available on crate feature tch only.

type Output = Vec<OsString>

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

type Output = Vec<&'a str>

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

Available on crate feature tch only.

type Output = Vec<&'a str>

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

type Output = Vec<&'a CStr>

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

Available on crate feature tch only.

type Output = Vec<&'a CStr>

impl<A> Collate<(A,)> for DefaultCollate

where A: Clone, DefaultCollate: Collate<A>,

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

impl<A> Collate<(A,)> for TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where 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 DefaultCollate

where 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 TorchCollate

where A: Clone + Element, D: Dimension, D::Larger: RemoveAxis,

Available on crate feature tch only.

type Output = Tensor

impl<K, V> Collate<BTreeMap<K, V>> for DefaultCollate

where K: Ord + Clone, V: Clone, Self: Collate<V>,

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

impl<K, V> Collate<BTreeMap<K, V>> for TorchCollate

where K: Ord + Clone, V: Clone, Self: Collate<V>,

Available on crate feature tch only.

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

impl<K, V, H> Collate<HashMap<K, V, H>> for DefaultCollate

where K: Eq + Hash + Clone, V: Clone, Self: Collate<V>, H: BuildHasher,

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

impl<K, V, H> Collate<HashMap<K, V, H>> for TorchCollate

where 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>

impl<T> Collate<&T> for DefaultCollate

where 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 TorchCollate

where 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>> for DefaultCollate

where Self: Collate<T>, T: Clone,

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

impl<T> Collate<VecDeque<T>> for TorchCollate

where Self: Collate<T>, T: Clone,

Available on crate feature tch only.

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

impl<T> Collate<Vec<T>> for DefaultCollate

where Self: Collate<T>, T: Clone,

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

impl<T> Collate<Vec<T>> for TorchCollate

where Self: Collate<T>, T: Clone,

Available on crate feature tch only.

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

impl<T> Collate<T> for NoOpCollate

type Output = Vec<T>

impl<T, F, O> Collate<T> for F

where F: Fn(Vec<T>) -> O,

type Output = O

impl<T, const N: usize> Collate<[T; N]> for DefaultCollate

where Self: Collate<T>, T: Clone,

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

impl<T, const N: usize> Collate<[T; N]> for TorchCollate

where Self: Collate<T>, T: Clone,

Available on crate feature tch only.

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