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 = bool

The type of the tensor elements.

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

Creates an empty tensor with the given shape.

fn register_transaction(tr: &mut Transaction<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, ranges: &[Range<usize>]) -> Self::Primitive

Select tensor elements corresponding for the given ranges.

fn slice_assign( tensor: Self::Primitive, ranges: &[Range<usize>], value: Self::Primitive, ) -> Self::Primitive

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

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) -> TensorData

Extracts the data from the tensor asynchronously.

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

Creates a tensor from the given data.

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 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 chunk( tensor: Self::Primitive, chunks: usize, dim: usize, ) -> Vec<Self::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 split( tensor: Self::Primitive, split_size: usize, dim: usize, ) -> Vec<Self::Primitive>

Splits the tensor into chunks of a specified size along a given dimension. Each chunk is a view of the original tensor.

fn split_with_sizes( tensor: Self::Primitive, split_sizes: Vec<usize>, dim: usize, ) -> Vec<Self::Primitive>

Splits the tensor into chunks with the specified sizes along a given dimension. Each chunk is a view of the original tensor.

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

The primitive type of the tensor.

fn name() -> &'static str

The name of the tensor kind.