Struct 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 = <B as Backend>::BoolElem

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 register_transaction( tr: &mut Transaction<B>, tensor: <Bool as TensorKind<B>>::Primitive, )

Read the data from the tensor using a transaction.

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 from_data_dtype( data: TensorData, device: &<B as Backend>::Device, dtype: DType, ) -> <Bool as TensorKind<B>>::Primitive

Creates a tensor from the given data enforcing the given data type.

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 slice_fill( tensor: Self::Primitive, ranges: &[Range<usize>], value: Self::Elem, ) -> Self::Primitive

Fills the tensor elements corresponding for the given ranges with the given value.

fn slice_dim( tensor: Self::Primitive, dim: usize, range: &Range<usize>, ) -> Self::Primitive

Slices the tensor along a given dimension with the specified range.

fn elem_type_name() -> &'static str

Returns the name of the element type.

fn dtype(tensor: &Self::Primitive) -> DType

Returns the tensor data type.

impl Clone for Bool

fn clone(&self) -> Bool

Returns a duplicate 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.