pub struct LinearBatchCrossEntropyMulticlassArgs<T>where
T: DataTypeInfo,{
pub actual: CudaPtr<T>,
/* private fields */
}Expand description
Defines the list passed to the cuda kernel function as the argument of croos entropy multiclass
Fields
actual: CudaPtr<T>actual value
Implementations
sourceimpl<T> LinearBatchCrossEntropyMulticlassArgs<T>where
T: DataTypeInfo,
impl<T> LinearBatchCrossEntropyMulticlassArgs<T>where
T: DataTypeInfo,
sourcepub fn new(
t: CudaPtr<T>,
r: CudaPtr<T>,
out_len: usize,
batch_len: usize
) -> LinearBatchCrossEntropyMulticlassArgs<T>
pub fn new(
t: CudaPtr<T>,
r: CudaPtr<T>,
out_len: usize,
batch_len: usize
) -> LinearBatchCrossEntropyMulticlassArgs<T>
Create a LinearBatchCrossEntropyMulticlassArgs instance
Arguments
expected- Expected Valueactual- Actual Valueout_len- Number of scalar values in outputbatch_len- batch count
Trait Implementations
sourceimpl<T> KernelArgs for LinearBatchCrossEntropyMulticlassArgs<T>where
T: DataTypeInfo,
impl<T> KernelArgs for LinearBatchCrossEntropyMulticlassArgs<T>where
T: DataTypeInfo,
sourcefn as_vec(&mut self) -> Vec<&mut dyn AsMutKernelPtr>
fn as_vec(&mut self) -> Vec<&mut dyn AsMutKernelPtr>
Returns a Vec<&mut dyn AsMutKernelPtr> of the type implementing AsMutKernelPtr,
which is converted to a data type that can be passed to the cuda kernel in subsequent processing. Read more
Auto Trait Implementations
impl<T> RefUnwindSafe for LinearBatchCrossEntropyMulticlassArgs<T>where
T: RefUnwindSafe,
impl<T> !Send for LinearBatchCrossEntropyMulticlassArgs<T>
impl<T> !Sync for LinearBatchCrossEntropyMulticlassArgs<T>
impl<T> Unpin for LinearBatchCrossEntropyMulticlassArgs<T>
impl<T> UnwindSafe for LinearBatchCrossEntropyMulticlassArgs<T>where
T: RefUnwindSafe,
Blanket Implementations
sourceimpl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
const: unstable · sourcefn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more