Struct burn_tensor::BoolTensor

pub struct BoolTensor<B: Backend, const D: usize> { /* private fields */ }

Implementations

impl<B, const D: usize> BoolTensor<B, D>where B: Backend,

pub fn new(tensor: B::BoolTensorPrimitive<D>) -> Self

pub fn shape(&self) -> Shape<D>

pub fn to_device(&self, device: &B::Device) -> Self

pub fn dims(&self) -> [usize; D]

Returns the dimensions of the current tensor.

Equivalent to tensor.shape().dims.

pub fn into_data(self) -> Data<bool, D>

pub fn to_data(&self) -> Data<bool, D>

pub fn from_data(data: Data<bool, D>) -> Self

pub fn to_int(&self) -> Tensor<B::IntegerBackend, D>

pub fn from_int_backend(tensor: BoolTensor<B::IntegerBackend, D>) -> Self

pub fn reshape<const D2: usize, S: Into<Shape<D2>>>( &self, shape: S ) -> BoolTensor<B, D2>

Reshape the tensor to have the given shape.

Panics

If the tensor can not be reshape to the given shape.

pub fn index<const D2: usize>(&self, indexes: [Range<usize>; D2]) -> Self

Trait Implementations

impl<B: Clone + Backend, const D: usize> Clone for BoolTensor<B, D>where B::BoolTensorPrimitive<D>: Clone,

fn clone(&self) -> BoolTensor<B, D>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<B: Debug + Backend, const D: usize> Debug for BoolTensor<B, D>where B::BoolTensorPrimitive<D>: Debug,

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

Formats the value using the given formatter.