pub struct TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S: State = Empty> { /* private fields */ }Expand description
Use builder syntax to set the inputs and finish with call().
Implementations§
Source§impl<'f1, 'f2, 'f3, 'f4, 'f5, S: State> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
impl<'f1, 'f2, 'f3, 'f4, 'f5, S: State> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
Sourcepub fn call(self) -> Result<Tensor>where
S: IsComplete,
pub fn call(self) -> Result<Tensor>where
S: IsComplete,
Finishes building and performs the requested action.
Sourcepub fn scale(
self,
value: &'f2 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetScale<S>>where
S::Scale: IsUnset,
pub fn scale(
self,
value: &'f2 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetScale<S>>where
S::Scale: IsUnset,
Sourcepub fn maybe_scale(
self,
value: Option<&'f2 Tensor>,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetScale<S>>where
S::Scale: IsUnset,
pub fn maybe_scale(
self,
value: Option<&'f2 Tensor>,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetScale<S>>where
S::Scale: IsUnset,
Sourcepub fn bias(
self,
value: &'f3 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetBias<S>>where
S::Bias: IsUnset,
pub fn bias(
self,
value: &'f3 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetBias<S>>where
S::Bias: IsUnset,
Sourcepub fn maybe_bias(
self,
value: Option<&'f3 Tensor>,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetBias<S>>where
S::Bias: IsUnset,
pub fn maybe_bias(
self,
value: Option<&'f3 Tensor>,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetBias<S>>where
S::Bias: IsUnset,
Sourcepub fn mean(
self,
value: &'f4 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetMean<S>>where
S::Mean: IsUnset,
pub fn mean(
self,
value: &'f4 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetMean<S>>where
S::Mean: IsUnset,
Required.
Sourcepub fn invstd(
self,
value: &'f5 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetInvstd<S>>where
S::Invstd: IsUnset,
pub fn invstd(
self,
value: &'f5 Tensor,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetInvstd<S>>where
S::Invstd: IsUnset,
Required.
Sourcepub fn axis(
self,
value: AxisSpec,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetAxis<S>>where
S::Axis: IsUnset,
pub fn axis(
self,
value: AxisSpec,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetAxis<S>>where
S::Axis: IsUnset,
Sourcepub fn maybe_axis(
self,
value: Option<AxisSpec>,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetAxis<S>>where
S::Axis: IsUnset,
pub fn maybe_axis(
self,
value: Option<AxisSpec>,
) -> TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, SetAxis<S>>where
S::Axis: IsUnset,
Auto Trait Implementations§
impl<'f1, 'f2, 'f3, 'f4, 'f5, S> Freeze for TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
impl<'f1, 'f2, 'f3, 'f4, 'f5, S = Empty> !RefUnwindSafe for TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
impl<'f1, 'f2, 'f3, 'f4, 'f5, S> Send for TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
impl<'f1, 'f2, 'f3, 'f4, 'f5, S> Sync for TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
impl<'f1, 'f2, 'f3, 'f4, 'f5, S> Unpin for TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
impl<'f1, 'f2, 'f3, 'f4, 'f5, S> UnsafeUnpin for TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
impl<'f1, 'f2, 'f3, 'f4, 'f5, S = Empty> !UnwindSafe for TensorBatchnormBuilder<'f1, 'f2, 'f3, 'f4, 'f5, S>
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
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more