pub struct TensorSpec {
pub name: String,
pub shape: Vec<usize>,
pub dtype: DataType,
pub device: Device,
pub source: WeightSource,
pub transformations: Vec<TensorTransformation>,
}Expand description
Tensor specification for weight loading
Fields§
§name: StringName/identifier of the tensor
shape: Vec<usize>Expected tensor shape
dtype: DataTypeTarget data type
device: DeviceTarget device
source: WeightSourceWeight source location
transformations: Vec<TensorTransformation>Optional transformations to apply
Trait Implementations§
Source§impl Clone for TensorSpec
impl Clone for TensorSpec
Source§fn clone(&self) -> TensorSpec
fn clone(&self) -> TensorSpec
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 TensorSpec
impl Debug for TensorSpec
Source§impl<'de> Deserialize<'de> for TensorSpec
impl<'de> Deserialize<'de> for TensorSpec
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for TensorSpec
impl RefUnwindSafe for TensorSpec
impl Send for TensorSpec
impl Sync for TensorSpec
impl Unpin for TensorSpec
impl UnsafeUnpin for TensorSpec
impl UnwindSafe for TensorSpec
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