Struct burn_core::tensor::Int

pub struct Int;

Trait Implementations

impl<B> BasicOps<B> for Intwhere B: Backend,

type Elem = <B as Backend>::IntElem

fn empty<const D: usize>( shape: Shape<D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>

fn shape<const D: usize>( tensor: &<Int as TensorKind<B>>::Primitive<D> ) -> Shape<D>

fn reshape<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, shape: Shape<D2> ) -> <Int as TensorKind<B>>::Primitive<D2>

fn index<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, indexes: [Range<usize>; D2] ) -> <Int as TensorKind<B>>::Primitive<D1>

fn index_assign<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, indexes: [Range<usize>; D2], value: <Int as TensorKind<B>>::Primitive<D1> ) -> <Int as TensorKind<B>>::Primitive<D1>

fn device<const D: usize>( tensor: &<Int as TensorKind<B>>::Primitive<D> ) -> <B as Backend>::Device

fn to_device<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>

fn into_data<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> Data<<Int as BasicOps<B>>::Elem, D>

fn from_data<const D: usize>( data: Data<<Int as BasicOps<B>>::Elem, D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>

fn repeat<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize, times: usize ) -> <Int as TensorKind<B>>::Primitive<D>

fn equal<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, D, Bool>

fn equal_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>

fn cat<const D: usize>( vectors: Vec<<Int as TensorKind<B>>::Primitive<D>, Global>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>

fn elem_type_name() -> &'static str

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<(), Error>

Formats the value using the given formatter.

impl<B> Numeric<B> for Intwhere B: Backend,

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

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

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

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

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

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

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

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

fn neg<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>

fn zeros<const D: usize>( shape: Shape<D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>

fn ones<const D: usize>( shape: Shape<D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>

fn sum<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<1>

fn sum_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>

fn mean<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<1>

fn mean_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>

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

fn greater_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>

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

fn greater_equal_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>

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

fn lower_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>

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

fn lower_equal_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>

fn mask_scatter<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, mask: Tensor<B, D, Bool>, source: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>

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

fn index_select_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize, indexes: Tensor<B, 1, Int> ) -> <Int as TensorKind<B>>::Primitive<D>

fn index_select_dim_assign<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, dim: usize, indexes: Tensor<B, 1, Int>, values: <Int as TensorKind<B>>::Primitive<D2> ) -> <Int as TensorKind<B>>::Primitive<D1>

fn index_select<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, indexes: Tensor<B, D, Int> ) -> <Int as TensorKind<B>>::Primitive<D>

fn index_select_assign<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, indexes: Tensor<B, D, Int>, values: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>

impl<B> TensorKind<B> for Intwhere B: Backend,

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

fn name() -> &'static str