Struct UpsampleLayer 

pub struct UpsampleLayer { /* private fields */ }

Upsample layer: nearest or bilinear upsampling.

Implementations

impl UpsampleLayer

pub fn new(scale_factor: usize, bilinear: bool) -> Self

pub fn scale_factor(&self) -> usize

pub fn is_bilinear(&self) -> bool

pub fn forward_inference(&self, input: &Tensor) -> Result<Tensor, ModelError>

pub fn forward( &self, graph: &mut Graph, input: NodeId, ) -> Result<NodeId, ModelError>

Trait Implementations

impl Clone for UpsampleLayer

fn clone(&self) -> UpsampleLayer

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for UpsampleLayer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for UpsampleLayer

fn eq(&self, other: &UpsampleLayer) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for UpsampleLayer

impl StructuralPartialEq for UpsampleLayer