Struct relearn::torch::packed::SharedBatchSizes

source · [−]

pub struct SharedBatchSizes { /* private fields */ }

Expand description

Slice of a reference-counted batch sizes vector.

The value of the ith index of the batch sizes vector is the number of sequences with length at least i + 1. It is the number of index-i steps that appear in the packed tensor.

Implementations

source

impl SharedBatchSizes

source

pub fn from_batch_sizes<I: IntoIterator<Item = usize>>(
batch_sizes: I
) -> Result<Self, PackingError>

Construct from an iterator of monotonic decreasing batch sizes.

Returns an error if any batch size is greater than the previous batch size.

source

pub fn from_sorted_sequence_lengths<I: IntoIterator<Item = usize>>(
lengths: I
) -> Result<Self, PackingError>

Construct from an iterator of monotonic decreasing sequence lengths.

Returns an error if any length is greater than the previous length.

source

pub fn as_slice(&self) -> &[i64]

View batch sizes as a slice

source

pub fn tensor(&self) -> Tensor

Batch size as an i64 [Tensor]. The underlying data is cached and shared between calls.

source

pub fn len(&self) -> usize

The total number of elements across all sequences represented by this structure.

source

pub fn is_empty(&self) -> bool

Whether the total number of elements across all sequences is zero.

source

pub const fn trim(self, n: usize) -> Self

Batch sizes resulting from removing n values from each sequence.

Trait Implementations

source

impl AsRef<[i64]> for SharedBatchSizes

source

fn as_ref(&self) -> &[i64]

Converts this type into a shared reference of the (usually inferred) input type.

source

impl Clone for SharedBatchSizes

source

fn clone(&self) -> SharedBatchSizes

Returns a copy of the value. Read more

1.0.0 · source

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more

source

impl Debug for SharedBatchSizes

source

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

source

impl<'de> Deserialize<'de> for SharedBatchSizes

source

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error> where
__D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer. Read more

source

impl<T: AsRef<[i64]>> PartialEq<T> for SharedBatchSizes

source

fn eq(&self, other: &T) -> bool

This method tests for self and other values to be equal, and is used by ==. Read more

1.0.0 · source

fn ne(&self, other: &Rhs) -> bool

This method tests for !=.

source

impl Serialize for SharedBatchSizes

source

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error> where
S: Serializer,

Serialize this value into the given Serde serializer. Read more

source

impl Eq for SharedBatchSizes

Auto Trait Implementations

impl RefUnwindSafe for SharedBatchSizes

impl !Send for SharedBatchSizes

impl !Sync for SharedBatchSizes

impl Unpin for SharedBatchSizes

impl UnwindSafe for SharedBatchSizes

Blanket Implementations

source

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

source

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

source

impl<T> AsAny for T where
T: Any,

source

fn as_any(&self) -> &(dyn Any + 'static)

Convert into an Any trait reference.

source

impl<T> Borrow<T> for T where
T: ?Sized,

const: unstable · source

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

source

impl<T> BorrowMut<T> for T where
T: ?Sized,

const: unstable · source

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

source

impl<Q, K> Equivalent<K> for Q where
Q: Eq + ?Sized,
K: Borrow<Q> + ?Sized,

source

fn equivalent(&self, key: &K) -> bool

Compare self to key and return true if they are equal.

source

impl<T> From<T> for T

const: unstable · source

fn from(t: T) -> T

Returns the argument unchanged.

source

impl<T, U> Into for T where
U: From<T>,

const: unstable · source

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Pointable for T

const ALIGN: usize = mem::align_of::<T>()

The alignment of pointer.

type Init = T

The type for initializers.

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more

source

impl<T> Same<T> for T

type Output = T

Should always be Self

source

impl<T> ToOwned for T where
T: Clone,

type Owned = T

The resulting type after obtaining ownership.

source

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more

source

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more

source

impl<T, U> TryFrom for T where
U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

const: unstable · source

fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>

Performs the conversion.

source

impl<T, U> TryInto for T where
U: TryFrom<T>,

type Error = >::Error

The type returned in the event of a conversion error.

const: unstable · source

fn try_into(self) -> Result<U, >::Error>

Performs the conversion.

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

fn vzip(self) -> V

source

Struct relearn::torch::packed::SharedBatchSizes

Implementations

impl SharedBatchSizes

pub fn from_batch_sizes<I: IntoIterator<Item = usize>>( batch_sizes: I) -> Result<Self, PackingError>

pub fn from_sorted_sequence_lengths<I: IntoIterator<Item = usize>>( lengths: I) -> Result<Self, PackingError>

pub fn as_slice(&self) -> &[i64]

pub fn tensor(&self) -> Tensor

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub const fn trim(self, n: usize) -> Self

Trait Implementations

impl AsRef<[i64]> for SharedBatchSizes

fn as_ref(&self) -> &[i64]

impl Clone for SharedBatchSizes

fn clone(&self) -> SharedBatchSizes

fn clone_from(&mut self, source: &Self)

impl Debug for SharedBatchSizes

fn fmt(&self, f: &mut Formatter<'_>) -> Result

impl<'de> Deserialize<'de> for SharedBatchSizes

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where __D: Deserializer<'de>,

impl<T: AsRef<[i64]>> PartialEq<T> for SharedBatchSizes

fn eq(&self, other: &T) -> bool

fn ne(&self, other: &Rhs) -> bool

impl Serialize for SharedBatchSizes

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error> where __S: Serializer,

impl Eq for SharedBatchSizes

Auto Trait Implementations

impl RefUnwindSafe for SharedBatchSizes

impl !Send for SharedBatchSizes

impl !Sync for SharedBatchSizes

impl Unpin for SharedBatchSizes

impl UnwindSafe for SharedBatchSizes

Blanket Implementations

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

fn type_id(&self) -> TypeId

impl<T> AsAny for T where T: Any,

fn as_any(&self) -> &(dyn Any + 'static)

impl<T> Borrow<T> for T where T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for T where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<Q, K> Equivalent<K> for Q where Q: Eq + ?Sized, K: Borrow<Q> + ?Sized,

fn equivalent(&self, key: &K) -> bool

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Into<U> for T where U: From<T>,

fn into(self) -> U

impl<T> Pointable for T

const ALIGN: usize = mem::align_of::<T>()

type Init = T

unsafe fn init(init: <T as Pointable>::Init) -> usize

unsafe fn deref<'a>(ptr: usize) -> &'a T

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

unsafe fn drop(ptr: usize)

impl<T> Same<T> for T

type Output = T

impl<T> ToOwned for T where T: Clone,

type Owned = T

fn to_owned(&self) -> T

fn clone_into(&self, target: &mut T)

impl<T, U> TryFrom<U> for T where U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for T where U: TryFrom<T>,

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

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

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

fn vzip(self) -> V

impl<T> DeserializeOwned for T where T: for<'de> Deserialize<'de>,

pub fn from_batch_sizes<I: IntoIterator<Item = usize>>(
batch_sizes: I
) -> Result<Self, PackingError>

pub fn from_sorted_sequence_lengths<I: IntoIterator<Item = usize>>(
lengths: I
) -> Result<Self, PackingError>

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error> where
__D: Deserializer<'de>,

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error> where
S: Serializer,

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

impl<T> AsAny for T where
T: Any,

impl<T> Borrow<T> for T where
T: ?Sized,

impl<T> BorrowMut<T> for T where
T: ?Sized,

impl<Q, K> Equivalent<K> for Q where
Q: Eq + ?Sized,
K: Borrow<Q> + ?Sized,

impl<T, U> Into<U> for T where
U: From<T>,

impl<T> ToOwned for T where
T: Clone,

impl<T, U> TryFrom<U> for T where
U: Into<T>,

impl<T, U> TryInto<U> for T where
U: TryFrom<T>,

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

impl<T> DeserializeOwned for T where
T: for<'de> Deserialize<'de>,