Struct burn_tensor::Bool

pub struct Bool;

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

Trait Implementations

impl<B: Backend> BasicOps<B> for Bool

type Elem = bool

The type of the tensor elements.

fn empty<const D: usize>( shape: Shape<D>, device: &B::Device ) -> Self::Primitive<D>

Creates an empty tensor with the given shape.

fn shape<const D: usize>(tensor: &Self::Primitive<D>) -> Shape<D>

Returns the shape of the tensor.

fn reshape<const D1: usize, const D2: usize>( tensor: Self::Primitive<D1>, shape: Shape<D2> ) -> Self::Primitive<D2>

Reshapes the tensor.

fn slice<const D1: usize, const D2: usize>( tensor: Self::Primitive<D1>, ranges: [Range<usize>; D2] ) -> Self::Primitive<D1>

Select tensor elements corresponding for the given ranges.

fn slice_assign<const D1: usize, const D2: usize>( tensor: Self::Primitive<D1>, ranges: [Range<usize>; D2], value: Self::Primitive<D1> ) -> Self::Primitive<D1>

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

fn device<const D: usize>(tensor: &Self::Primitive<D>) -> <B as Backend>::Device

Returns the device on which the tensor is allocated.

fn to_device<const D: usize>( tensor: Self::Primitive<D>, device: &<B as Backend>::Device ) -> Self::Primitive<D>

Moves the tensor to the given device.

fn into_data<const D: usize>(tensor: Self::Primitive<D>) -> Data<Self::Elem, D>

Extracts the data from the tensor.

fn from_data<const D: usize>( data: Data<Self::Elem, D>, device: &B::Device ) -> Self::Primitive<D>

Creates a tensor from the given data.

fn repeat<const D: usize>( tensor: Self::Primitive<D>, dim: usize, times: usize ) -> Self::Primitive<D>

Repeat the tensor along the given dimension.

fn equal<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D> ) -> Tensor<B, D, Bool>

Equates the given tensors.

fn cat<const D: usize>( vectors: Vec<Self::Primitive<D>>, dim: usize ) -> Self::Primitive<D>

Concatenates the given tensors along the given dimension.

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

Formats the value using the given formatter.

impl<B: Backend> TensorKind<B> for Bool

type Primitive<const D: usize> = <B as Backend>::BoolTensorPrimitive<D>

The primitive type of the tensor.

fn name() -> &'static str

The name of the tensor kind.