Struct Bool 

pub struct Bool;

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

Trait Implementations

impl<B: AutodiffBackend> BasicAutodiffOps<B> for Bool

type InnerKind = Bool

Inner primitive tensor.

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

Returns the inner tensor without the autodiff information.

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

Convert a tensor to the autodiff backend.

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

type Elem = <B as Backend>::BoolElem

The type of the tensor elements.

fn empty(shape: Shape, device: &Device<B>, dtype: DType) -> Self::Primitive

Creates an empty tensor with the given shape.

fn zeros(shape: Shape, device: &Device<B>, dtype: DType) -> Self::Primitive

Creates a tensor filled with zeros.

fn ones(shape: Shape, device: &Device<B>, dtype: DType) -> Self::Primitive

Creates a tensor filled with ones.

fn full<E: ElementConversion>( shape: Shape, fill_value: E, device: &Device<B>, dtype: DType, ) -> Self::Primitive

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

fn register_transaction( tr: &mut TransactionPrimitive<B>, tensor: Self::Primitive, )

Read the data from the tensor using a transaction.

fn reshape(tensor: Self::Primitive, shape: Shape) -> Self::Primitive

Reshapes the tensor.

fn transpose(tensor: Self::Primitive) -> Self::Primitive

Transposes a tensor.

fn swap_dims( tensor: Self::Primitive, dim1: usize, dim2: usize, ) -> Self::Primitive

Swaps two dimensions of a tensor.

fn slice(tensor: Self::Primitive, slices: &[Slice]) -> Self::Primitive

Select tensor elements corresponding to the given slices.

fn slice_assign( tensor: Self::Primitive, slices: &[Slice], value: Self::Primitive, ) -> Self::Primitive

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

fn select( tensor: Self::Primitive, dim: usize, indices: IntTensor<B>, ) -> Self::Primitive

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

fn select_assign( tensor: Self::Primitive, dim: usize, indices: IntTensor<B>, values: Self::Primitive, update: IndexingUpdateOp, ) -> Self::Primitive

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

fn mask_where( tensor: Self::Primitive, mask: B::BoolTensorPrimitive, source: Self::Primitive, ) -> Self::Primitive

Selects elements from a tensor based on a boolean mask.

fn mask_fill( tensor: Self::Primitive, mask: B::BoolTensorPrimitive, value: Self::Elem, ) -> Self::Primitive

Fills elements of a tensor based on a boolean mask.

fn gather( dim: usize, tensor: Self::Primitive, indices: B::IntTensorPrimitive, ) -> Self::Primitive

Gathers elements from a tensor along an axis.

fn scatter( dim: usize, tensor: Self::Primitive, indices: B::IntTensorPrimitive, values: Self::Primitive, update: IndexingUpdateOp, ) -> Self::Primitive

Scatters elements into a tensor along an axis.

fn device(tensor: &Self::Primitive) -> Device<B>

Returns the device on which the tensor is allocated.

fn to_device(tensor: Self::Primitive, device: &Device<B>) -> Self::Primitive

Moves the tensor to the given device.

async fn into_data_async( tensor: Self::Primitive, ) -> Result<TensorData, ExecutionError>

Extracts the data from the tensor asynchronously.

fn from_data(data: TensorData, device: &Device<B>) -> Self::Primitive

Creates a tensor from the given data.

fn from_data_dtype( data: TensorData, device: &Device<B>, dtype: DType, ) -> Self::Primitive

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

fn repeat_dim( tensor: Self::Primitive, dim: usize, times: usize, ) -> Self::Primitive

Repeat the tensor along the given dimension.

fn equal(lhs: Self::Primitive, rhs: Self::Primitive) -> B::BoolTensorPrimitive

Equates the given tensors.

fn not_equal( lhs: Self::Primitive, rhs: Self::Primitive, ) -> B::BoolTensorPrimitive

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

fn equal_elem(lhs: Self::Primitive, rhs: Self::Elem) -> B::BoolTensorPrimitive

Element-wise equality between two tensors.

fn not_equal_elem( lhs: Self::Primitive, rhs: Self::Elem, ) -> B::BoolTensorPrimitive

Element-wise non-equality between two tensors.

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

Concatenates the given tensors along the given dimension.

fn any(tensor: Self::Primitive) -> B::BoolTensorPrimitive

Tests if any element in the tensor evaluates to True.

fn any_dim(tensor: Self::Primitive, dim: usize) -> B::BoolTensorPrimitive

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

fn all(tensor: Self::Primitive) -> B::BoolTensorPrimitive

Tests if all elements in the tensor evaluate to True.

fn all_dim(tensor: Self::Primitive, dim: usize) -> B::BoolTensorPrimitive

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

fn permute(tensor: Self::Primitive, axes: &[usize]) -> Self::Primitive

Permutes the dimensions of a tensor.

fn expand(tensor: Self::Primitive, shape: Shape) -> Self::Primitive

Broadcasts the given tensor to the specified shape.

fn flip(tensor: Self::Primitive, axes: &[usize]) -> Self::Primitive

Flips the tensor along the given axes.

fn unfold( tensor: Self::Primitive, dim: usize, size: usize, step: usize, ) -> Self::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

Formats the value using the given formatter.

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

type Primitive = <B as Backend>::BoolTensorPrimitive

The primitive type of the tensor.

fn name() -> &'static str

The name of the tensor kind.