pub struct GridSampleArgs<'a, T: Element> {
pub input: TensorRef<'a, T, 4>,
pub grid: TensorRef<'a, T, 4>,
pub output: TensorMut<'a, T, 4>,
}Expand description
Args bundle for a grid_sample launch.
Fields§
§input: TensorRef<'a, T, 4>Input [N, C, IH, IW]. NCHW row-major contiguous.
grid: TensorRef<'a, T, 4>Sampling grid [N, OH, OW, 2] — (x, y) normalized in [-1, 1].
output: TensorMut<'a, T, 4>Output [N, C, OH, OW].
Auto Trait Implementations§
impl<'a, T> !UnwindSafe for GridSampleArgs<'a, T>
impl<'a, T> Freeze for GridSampleArgs<'a, T>
impl<'a, T> RefUnwindSafe for GridSampleArgs<'a, T>where
T: RefUnwindSafe,
impl<'a, T> Send for GridSampleArgs<'a, T>
impl<'a, T> Sync for GridSampleArgs<'a, T>where
T: Sync,
impl<'a, T> Unpin for GridSampleArgs<'a, T>
impl<'a, T> UnsafeUnpin for GridSampleArgs<'a, T>
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