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, dtype: DType, ) -> <Bool as TensorKind<B>>::Primitive

Creates an empty tensor with the given shape.

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

Creates a tensor filled with zeros.

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

Creates a tensor filled with ones.

fn full<E>( shape: Shape, fill_value: E, device: &<B as Backend>::Device, dtype: DType, ) -> <Bool as TensorKind<B>>::Primitive

where E: ElementConversion,

Creates a tensor of the given shape where each element is equal to the provided value.

fn register_transaction( tr: &mut TransactionPrimitive<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, slices: &[Slice], ) -> <Bool as TensorKind<B>>::Primitive

Select tensor elements corresponding to the given slices.

fn slice_assign( tensor: <Bool as TensorKind<B>>::Primitive, slices: &[Slice], value: <Bool as TensorKind<B>>::Primitive, ) -> <Bool as TensorKind<B>>::Primitive

Assigns the given value to the tensor elements corresponding to the given slices.

fn select( tensor: <Bool as TensorKind<B>>::Primitive, dim: usize, indices: <B as Backend>::IntTensorPrimitive, ) -> <Bool as TensorKind<B>>::Primitive

Select tensor elements along the given dimension corresponding to the given indices.

fn select_assign( tensor: <Bool as TensorKind<B>>::Primitive, dim: usize, indices: <B as Backend>::IntTensorPrimitive, values: <Bool as TensorKind<B>>::Primitive, update: IndexingUpdateOp, ) -> <Bool as TensorKind<B>>::Primitive

Assign the selected elements along the given dimension corresponding to the given indices from the value tensor.

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

Selects elements from a tensor based on a boolean mask.

fn mask_fill( tensor: <Bool as TensorKind<B>>::Primitive, mask: <B as Backend>::BoolTensorPrimitive, value: <Bool as BasicOps<B>>::Elem, ) -> <Bool as TensorKind<B>>::Primitive

Fills elements of a tensor based on a boolean mask.

fn gather( dim: usize, tensor: <Bool as TensorKind<B>>::Primitive, indices: <B as Backend>::IntTensorPrimitive, ) -> <Bool as TensorKind<B>>::Primitive

Gathers elements from a tensor along an axis.

fn scatter( dim: usize, tensor: <Bool as TensorKind<B>>::Primitive, indices: <B as Backend>::IntTensorPrimitive, values: <Bool as TensorKind<B>>::Primitive, update: IndexingUpdateOp, ) -> <Bool as TensorKind<B>>::Primitive

Scatters elements into a tensor along an axis.

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, ) -> Result<TensorData, ExecutionError>

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 equal_elem( lhs: <Bool as TensorKind<B>>::Primitive, rhs: <Bool as BasicOps<B>>::Elem, ) -> <B as Backend>::BoolTensorPrimitive

Element-wise equality between two tensors.

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

Element-wise non-equality between two 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 unfold( tensor: <Bool as TensorKind<B>>::Primitive, dim: usize, size: usize, step: usize, ) -> <Bool as TensorKind<B>>::Primitive

Unfold windows along a dimension.

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.