Struct Mask 

pub struct Mask<B: Backend>(pub BoolTensor<B>);

A boolean mask primitive used with F::mask_fill.

Mirrors the slice of the bool-tensor API needed to build and broadcast the causal masks in the custom backward (construct → reshape → expand).

Tuple Fields

0: BoolTensor<B>

Implementations

impl<B: Backend> Mask<B>

pub fn tril_mask( rows: usize, cols: usize, offset: i64, device: &Device<B>, ) -> Self

[rows, cols] mask that is true strictly above the offset diagonal.

Matches Tensor::tril_mask: the true entries are the region a tril would fill.

pub fn reshape<const N: usize>(self, shape: [usize; N]) -> Self

Reshape the mask to a new rank.

pub fn expand<const N: usize>(self, shape: [usize; N]) -> Self

Broadcast-expand the mask.

Trait Implementations

impl<B: Backend> Clone for Mask<B>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.