Struct burn::prelude::Bool

pub struct Bool;

A type-level representation of the kind of a bool tensor.

Trait Implementations

impl<B> BasicAutodiffOps<B> for Bool

where B: AutodiffBackend,

type InnerKind = Bool

Inner primitive tensor.

fn inner( tensor: <Bool as TensorKind<B>>::Primitive, ) -> <<Bool as BasicAutodiffOps<B>>::InnerKind as TensorKind<<B as AutodiffBackend>::InnerBackend>>::Primitive

Returns the inner tensor without the autodiff information.

fn from_inner( inner: <<Bool as BasicAutodiffOps<B>>::InnerKind as TensorKind<<B as AutodiffBackend>::InnerBackend>>::Primitive, ) -> <Bool as TensorKind<B>>::Primitive

Convert a tensor to the autodiff backend.

impl<B> BasicOps<B> for Bool

where B: Backend,

type Elem = bool

The type of the tensor elements.

fn empty( shape: Shape, device: &<B as Backend>::Device, ) -> <Bool as TensorKind<B>>::Primitive

Creates an empty tensor with the given shape.

fn shape(tensor: &<Bool as TensorKind<B>>::Primitive) -> Shape

Returns the shape of the tensor.

fn reshape( tensor: <Bool as TensorKind<B>>::Primitive, shape: Shape, ) -> <Bool as TensorKind<B>>::Primitive

Reshapes the tensor.

fn transpose( tensor: <Bool as TensorKind<B>>::Primitive, ) -> <Bool as TensorKind<B>>::Primitive

Transposes a tensor.

fn swap_dims( tensor: <Bool as TensorKind<B>>::Primitive, dim1: usize, dim2: usize, ) -> <Bool as TensorKind<B>>::Primitive

Swaps two dimensions of a tensor.

fn slice( tensor: <Bool as TensorKind<B>>::Primitive, ranges: &[Range<usize>], ) -> <Bool as TensorKind<B>>::Primitive

Select tensor elements corresponding for the given ranges.

fn slice_assign( tensor: <Bool as TensorKind<B>>::Primitive, ranges: &[Range<usize>], value: <Bool as TensorKind<B>>::Primitive, ) -> <Bool as TensorKind<B>>::Primitive

Assigns the given value to the tensor elements corresponding for the given ranges.

fn device(tensor: &<Bool as TensorKind<B>>::Primitive) -> <B as Backend>::Device

Returns the device on which the tensor is allocated.

fn to_device( tensor: <Bool as TensorKind<B>>::Primitive, device: &<B as Backend>::Device, ) -> <Bool as TensorKind<B>>::Primitive

Moves the tensor to the given device.

async fn into_data_async( tensor: <Bool as TensorKind<B>>::Primitive, ) -> TensorData

Extracts the data from the tensor asynchronously.

fn from_data( data: TensorData, device: &<B as Backend>::Device, ) -> <Bool as TensorKind<B>>::Primitive

Creates a tensor from the given data.

fn repeat_dim( tensor: <Bool as TensorKind<B>>::Primitive, dim: usize, times: usize, ) -> <Bool as TensorKind<B>>::Primitive

Repeat the tensor along the given dimension.

fn equal( lhs: <Bool as TensorKind<B>>::Primitive, rhs: <Bool as TensorKind<B>>::Primitive, ) -> <B as Backend>::BoolTensorPrimitive

Equates the given tensors.

fn not_equal( lhs: <Bool as TensorKind<B>>::Primitive, rhs: <Bool as TensorKind<B>>::Primitive, ) -> <B as Backend>::BoolTensorPrimitive

Applies element-wise non-equality comparison between the given tensors.

fn cat( vectors: Vec<<Bool as TensorKind<B>>::Primitive>, dim: usize, ) -> <Bool as TensorKind<B>>::Primitive

Concatenates the given tensors along the given dimension.

fn any( tensor: <Bool as TensorKind<B>>::Primitive, ) -> <B as Backend>::BoolTensorPrimitive

Tests if any element in the tensor evaluates to True.

fn any_dim( tensor: <Bool as TensorKind<B>>::Primitive, dim: usize, ) -> <B as Backend>::BoolTensorPrimitive

Tests if any element in the tensor evaluates to True along a given dimension dim.

fn all( tensor: <Bool as TensorKind<B>>::Primitive, ) -> <B as Backend>::BoolTensorPrimitive

Tests if all elements in the tensor evaluate to True.

fn all_dim( tensor: <Bool as TensorKind<B>>::Primitive, dim: usize, ) -> <B as Backend>::BoolTensorPrimitive

Tests if all elements in the tensor evaluate to True along a given dimension dim.

fn permute( tensor: <Bool as TensorKind<B>>::Primitive, axes: &[usize], ) -> <Bool as TensorKind<B>>::Primitive

Permutes the dimensions of a tensor.

fn expand( tensor: <Bool as TensorKind<B>>::Primitive, shape: Shape, ) -> <Bool as TensorKind<B>>::Primitive

Broadcasts the given tensor to the specified shape.

fn flip( tensor: <Bool as TensorKind<B>>::Primitive, axes: &[usize], ) -> <Bool as TensorKind<B>>::Primitive

Flips the tensor along the given axes.

fn chunk( tensor: <Bool as TensorKind<B>>::Primitive, chunks: usize, dim: usize, ) -> Vec<<Bool as TensorKind<B>>::Primitive>

Attempts to split the tensor along the given dimension into chunks. May return less chunks than requested if the tensor size is not divisible by the number of chunks.

fn elem_type_name() -> &'static str

Returns the name of the element type.

impl Clone for Bool

fn clone(&self) -> Bool

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Bool

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

Formats the value using the given formatter.

impl<B> TensorKind<B> for Bool

where B: Backend,

type Primitive = <B as Backend>::BoolTensorPrimitive

The primitive type of the tensor.

fn name() -> &'static str

The name of the tensor kind.