pub struct TensorConfig {
pub enable_autograd: bool,
pub default_device: TensorDevice,
pub memory_layout: MemoryLayout,
pub enable_optimization: bool,
pub max_batch_size: usize,
}Expand description
Tensor configuration
Fields§
§enable_autograd: boolEnable automatic differentiation
default_device: TensorDeviceDefault tensor device (CPU or GPU)
memory_layout: MemoryLayoutMemory layout preference
enable_optimization: boolEnable graph optimization
max_batch_size: usizeMaximum tensor size for automatic batching
Trait Implementations§
Source§impl Clone for TensorConfig
impl Clone for TensorConfig
Source§fn clone(&self) -> TensorConfig
fn clone(&self) -> TensorConfig
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for TensorConfig
impl Debug for TensorConfig
Auto Trait Implementations§
impl Freeze for TensorConfig
impl RefUnwindSafe for TensorConfig
impl Send for TensorConfig
impl Sync for TensorConfig
impl Unpin for TensorConfig
impl UnwindSafe for TensorConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more