pub struct DropoutArgs<'a, T: Element, const N: usize> {
pub x: TensorRef<'a, T, N>,
pub y: TensorMut<'a, T, N>,
pub mask: TensorMut<'a, Bool, N>,
}Expand description
Args bundle for a dropout forward launch.
Fields§
§x: TensorRef<'a, T, N>Input tensor.
y: TensorMut<'a, T, N>Output tensor — same shape as x.
mask: TensorMut<'a, Bool, N>Mask tensor — packed Bool, same shape as x. Caller saves this
for DropoutBackwardArgs::mask.
Auto Trait Implementations§
impl<'a, T, const N: usize> !UnwindSafe for DropoutArgs<'a, T, N>
impl<'a, T, const N: usize> Freeze for DropoutArgs<'a, T, N>
impl<'a, T, const N: usize> RefUnwindSafe for DropoutArgs<'a, T, N>where
T: RefUnwindSafe,
impl<'a, T, const N: usize> Send for DropoutArgs<'a, T, N>
impl<'a, T, const N: usize> Sync for DropoutArgs<'a, T, N>where
T: Sync,
impl<'a, T, const N: usize> Unpin for DropoutArgs<'a, T, N>
impl<'a, T, const N: usize> UnsafeUnpin for DropoutArgs<'a, T, N>
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