Struct burn_tensor::Int

pub struct Int;

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

Trait Implementations

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

type InnerKind = Int

Inner primitive tensor.

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

Returns the inner tensor without the autodiff information.

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

Convert a tensor to the autodiff backend.

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

type Elem = <B as Backend>::IntElem

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 transpose<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<D>

Transposes a tensor.

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

Swaps two dimensions of a 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.

async fn into_data_async<const D: usize>( tensor: Self::Primitive<D>, ) -> TensorData

Extracts the data from the tensor asynchronously.

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

Creates a tensor from the given data.

fn repeat_dim<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 not_equal<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D>, ) -> Tensor<B, D, Bool>

Applies element-wise non-equality comparison between 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 any<const D: usize>(tensor: Self::Primitive<D>) -> Tensor<B, 1, Bool>

Tests if any element in the tensor evaluates to True.

fn any_dim<const D: usize>( tensor: Self::Primitive<D>, dim: usize, ) -> Tensor<B, D, Bool>

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

fn all<const D: usize>(tensor: Self::Primitive<D>) -> Tensor<B, 1, Bool>

Tests if all elements in the tensor evaluate to True.

fn all_dim<const D: usize>( tensor: Self::Primitive<D>, dim: usize, ) -> Tensor<B, D, Bool>

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

fn permute<const D: usize>( tensor: Self::Primitive<D>, axes: [usize; D], ) -> Self::Primitive<D>

Permutes the dimensions of a tensor.

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

Broadcasts the given tensor to the specified shape.

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

Flips the tensor along the given axes.

fn elem_type_name() -> &'static str

Returns the name of the element type.

impl Clone for Int

fn clone(&self) -> Int

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Int

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<B: Backend> Numeric<B> for Int

fn add<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D>, ) -> <Int as TensorKind<B>>::Primitive<D>

Adds two tensors together.

fn add_scalar<const D: usize, E: ElementConversion>( lhs: Self::Primitive<D>, rhs: E, ) -> Self::Primitive<D>

Adds a scalar to a tensor element-wise.

fn sub<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D>, ) -> <Int as TensorKind<B>>::Primitive<D>

Subtracts two tensors.

fn sub_scalar<const D: usize, E: ElementConversion>( lhs: Self::Primitive<D>, rhs: E, ) -> Self::Primitive<D>

Subtracts a scalar from a tensor element-wise.

fn div<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D>, ) -> <Int as TensorKind<B>>::Primitive<D>

Divides two tensors.

fn div_scalar<const D: usize, E: ElementConversion>( lhs: Self::Primitive<D>, rhs: E, ) -> Self::Primitive<D>

Divides a tensor by a scalar element-wise.

fn remainder_scalar<const D: usize, E: ElementConversion>( lhs: Self::Primitive<D>, rhs: E, ) -> Self::Primitive<D>

Computes the modulus element-wise. The result has the same sign as the divisor rhs and its absolute value is less than that of the divisor.

fn mul<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D>, ) -> <Int as TensorKind<B>>::Primitive<D>

Multiplies two tensors.

fn mul_scalar<const D: usize, E: ElementConversion>( lhs: Self::Primitive<D>, rhs: E, ) -> Self::Primitive<D>

Multiplies a tensor by a scalar element-wise.

fn neg<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<D>

Negates a tensor.

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

Creates a tensor filled with zeros.

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

Creates a tensor filled with ones.

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

Creates a tensor filled with elements equal to the given value.

fn sum<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<1>

Sums all the elements of the tensor.

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

Sums all the elements of the tensor along a dimension.

fn prod<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<1>

Computes the product of all the elements of the tensor.

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

Computes the product of all the elements of the tensor along a dimension.

fn mean<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<1>

Computes the mean of all the elements of the tensor.

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

Computes the mean of all the elements of the tensor along a dimension.

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

Element-wise equality between two tensors.

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

Element-wise non-equality between two tensors.

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

Element-wise greater than comparison between two tensors.

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

Element-wise greater than comparison between a tensor and a scalar.

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

Element-wise greater than or equal comparison between two tensors.

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

Element-wise greater than or equal comparison between a tensor and a scalar.

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

Element-wise less than comparison between two tensors.

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

Element-wise less than comparison between a tensor and a scalar.

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

Element-wise less than or equal comparison between two tensors.

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

Element-wise less than or equal comparison between a tensor and a scalar.

fn mask_where<const D: usize>( tensor: Self::Primitive<D>, mask: Tensor<B, D, Bool>, source: Self::Primitive<D>, ) -> Self::Primitive<D>

Selects elements from a tensor based on a boolean mask.

fn mask_fill<const D: usize>( tensor: Self::Primitive<D>, mask: Tensor<B, D, Bool>, value: Self::Elem, ) -> Self::Primitive<D>

Fills elements of a tensor based on a boolean mask.

fn select<const D: usize>( tensor: Self::Primitive<D>, dim: usize, indices: Tensor<B, 1, Int>, ) -> Self::Primitive<D>

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

fn select_assign<const D: usize>( tensor: Self::Primitive<D>, dim: usize, indices: Tensor<B, 1, Int>, values: Self::Primitive<D>, ) -> Self::Primitive<D>

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

fn gather<const D: usize>( dim: usize, tensor: Self::Primitive<D>, indices: Tensor<B, D, Int>, ) -> Self::Primitive<D>

Gathers elements from a tensor along an axis.

fn scatter<const D: usize>( dim: usize, tensor: Self::Primitive<D>, indices: Tensor<B, D, Int>, values: Self::Primitive<D>, ) -> Self::Primitive<D>

Scatters elements into a tensor along an axis.

fn argmax<const D: usize>( tensor: Self::Primitive<D>, dim: usize, ) -> <B as Backend>::IntTensorPrimitive<D>

Gets the indices of the maximum elements of a tensor along an axis.

fn argmin<const D: usize>( tensor: Self::Primitive<D>, dim: usize, ) -> <B as Backend>::IntTensorPrimitive<D>

Gets the indices of the minimum elements of a tensor along an axis.

fn max<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<1>

Gets the maximum elements of a tensor along an axis.

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

Gets the maximum elements of a tensor along an axis.

fn max_dim_with_indices<const D: usize>( tensor: Self::Primitive<D>, dim: usize, ) -> (Self::Primitive<D>, <B as Backend>::IntTensorPrimitive<D>)

Gets the maximum elements of a tensor along an axis.

fn min<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<1>

Gets the minimum elements of a tensor along an axis.

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

Gets the minimum elements of a tensor along an axis.

fn min_dim_with_indices<const D: usize>( tensor: Self::Primitive<D>, dim: usize, ) -> (Self::Primitive<D>, <B as Backend>::IntTensorPrimitive<D>)

Gets the minimum elements and indices of a tensor along an axis.

fn clamp<const D: usize>( tensor: Self::Primitive<D>, min: B::IntElem, max: B::IntElem, ) -> Self::Primitive<D>

Clamp the tensor between the given min and max values.

fn clamp_min<const D: usize>( tensor: Self::Primitive<D>, min: B::IntElem, ) -> Self::Primitive<D>

Clamps a tensor under a minimum value.

fn clamp_max<const D: usize>( tensor: Self::Primitive<D>, max: B::IntElem, ) -> Self::Primitive<D>

Clamps a tensor over a maximum value.

fn abs<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<D>

Calculate absolute value on all elements of a tensor

fn powf<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D>, ) -> Self::Primitive<D>

Element-wise power of a tensor to a float tensor

fn powf_scalar<const D: usize, E: ElementConversion>( lhs: Self::Primitive<D>, rhs: E, ) -> <Int as TensorKind<B>>::Primitive<D>

Element-wise power of a tensor to a scalar float

fn powi<const D: usize>( lhs: Self::Primitive<D>, rhs: Self::Primitive<D>, ) -> Self::Primitive<D>

Element-wise power of a tensor

fn powi_scalar<const D: usize, E: ElementConversion>( lhs: Self::Primitive<D>, rhs: E, ) -> Self::Primitive<D>

Element-wise power of a tensor to a scalar int

fn random<const D: usize>( shape: Shape<D>, distribution: Distribution, device: &<B as Backend>::Device, ) -> Self::Primitive<D>

Create a random tensor.

fn sign<const D: usize>(tensor: Self::Primitive<D>) -> Self::Primitive<D>

Returns the signs of the elements of a tensor.

fn sort<const D: usize>( tensor: Self::Primitive<D>, dim: usize, descending: bool, ) -> Self::Primitive<D>

Sort the elements of the input tensor by value along a given dimension.

fn sort_with_indices<const D: usize>( tensor: Self::Primitive<D>, dim: usize, descending: bool, ) -> (Self::Primitive<D>, <Int as TensorKind<B>>::Primitive<D>)

Sort the elements of the input tensor by value along a given dimension.

fn argsort<const D: usize>( tensor: Self::Primitive<D>, dim: usize, descending: bool, ) -> <Int as TensorKind<B>>::Primitive<D>

Returns the indices that sort the elements of the input tensor by value along a given dimension.

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

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

The primitive type of the tensor.

fn name() -> &'static str

The name of the tensor kind.