Struct Int 

pub struct Int;

Trait Implementations

impl<D: Device> BytesDTypeKind<D> for Int

fn from_bytes_dispatch( bytes: &[u8], shape: &Shape, device: &D, dtype: IntDType, ) -> Result<Self::Storage>

fn to_bytes_dispatch<'a>( storage: &'a Self::Storage, layout: &Layout, ) -> Result<Cow<'a, [u8]>>

impl<D: Device> ConstructDTypeKind<D> for Int

fn zeros_dispatch( shape: &Shape, device: &D, dtype: IntDType, ) -> Result<Self::Storage>

fn ones_dispatch( shape: &Shape, device: &D, dtype: IntDType, ) -> Result<Self::Storage>

fn full_dispatch( shape: &Shape, value: Self::Scalar, device: &D, dtype: IntDType, ) -> Result<Self::Storage>

impl<D: Device> DTypeKind<D> for Int

const KIND: KindTag = KindTag::Int

Runtime discriminant for this kind.

type Scalar = i64

type Storage = <D as Device>::IntStorage

type Meta = ()

type DType = IntDType

impl<D: Device> IndexingDTypeKind<D> for Int

fn index_select_dispatch( x: &Self::Storage, x_l: &Layout, idx: &D::IntStorage, idx_l: &Layout, dim: usize, ) -> Result<(Self::Storage, Shape)>

fn gather_dispatch( x: &Self::Storage, x_l: &Layout, idx: &D::IntStorage, idx_l: &Layout, dim: usize, ) -> Result<(Self::Storage, Shape)>

fn index_add_dispatch( init: &Self::Storage, init_l: &Layout, idx: &D::IntStorage, idx_l: &Layout, src: &Self::Storage, src_l: &Layout, dim: usize, ) -> Result<Self::Storage>

fn scatter_add_dispatch( init: &Self::Storage, init_l: &Layout, idx: &D::IntStorage, idx_l: &Layout, src: &Self::Storage, src_l: &Layout, dim: usize, ) -> Result<Self::Storage>

impl<D: Device, T: IntFrom> IntoTensor<D, Int> for T

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device, T: IntFrom> IntoTensor<D, Int> for &[T]

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device, T: IntFrom> IntoTensor<D, Int> for Vec<T>

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device, T: IntFrom, const N: usize> IntoTensor<D, Int> for [T; N]

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device, T: IntFrom, const N: usize> IntoTensor<D, Int> for &[T; N]

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device, T: IntFrom, const R: usize, const C: usize> IntoTensor<D, Int> for &[[T; C]; R]

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device, T: IntFrom, const D1: usize, const D2: usize, const D3: usize> IntoTensor<D, Int> for &[[[T; D3]; D2]; D1]

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device, T: IntFrom, const D1: usize, const D2: usize, const D3: usize, const D4: usize> IntoTensor<D, Int> for &[[[[T; D4]; D3]; D2]; D1]

fn shape(&self) -> Result<Shape>

fn into_storage(self, device: &D) -> Result<D::IntStorage>

impl<D: Device> NumericDTypeKind<D> for Int

fn binary_dispatch( lhs: &Self::Storage, lhs_l: &Layout, rhs: &Self::Storage, rhs_l: &Layout, op: BinaryOp, ) -> Result<Self::Storage>

fn binary_inplace_dispatch( dst: &mut Self::Storage, dst_l: &Layout, src: &Self::Storage, src_l: &Layout, op: BinaryOp, ) -> Result<()>

fn binary_scalar_dispatch( lhs: &Self::Storage, lhs_l: &Layout, rhs: Self::Scalar, op: BinaryOp, ) -> Result<Self::Storage>

fn binary_scalar_inplace_dispatch( dst: &mut Self::Storage, dst_l: &Layout, rhs: Self::Scalar, op: BinaryOp, ) -> Result<()>

fn binary_scalar_lhs_dispatch( scalar: Self::Scalar, rhs: &Self::Storage, rhs_l: &Layout, op: BinaryOp, ) -> Result<Self::Storage>

fn cmp_dispatch( lhs: &Self::Storage, lhs_l: &Layout, rhs: &Self::Storage, rhs_l: &Layout, op: CmpOp, ) -> Result<D::BoolStorage>

fn cmp_scalar_dispatch( lhs: &Self::Storage, lhs_l: &Layout, rhs: Self::Scalar, op: CmpOp, ) -> Result<D::BoolStorage>

fn unary_dispatch( x: &Self::Storage, l: &Layout, op: UnaryOp<Self::Scalar>, ) -> Result<Self::Storage>

fn unary_inplace_dispatch( dst: &mut Self::Storage, dst_l: &Layout, op: UnaryOp<Self::Scalar>, ) -> Result<()>

impl<D: Device> ReduceDTypeKind<D> for Int

fn arg_reduce_dispatch( x: &Self::Storage, layout: &Layout, dim: usize, keepdim: bool, take_max: bool, ) -> Result<(D::IntStorage, Shape)>

fn reduce_dispatch( x: &Self::Storage, l: &Layout, dims: &[usize], keepdim: bool, op: ReduceOp, ) -> Result<(Self::Storage, Shape)>

impl<D: Device> ShapeDTypeKind<D> for Int

fn contiguous_dispatch(s: &Self::Storage, l: &Layout) -> Result<Self::Storage>

fn cat_dispatch( srcs: &[(&Self::Storage, &Layout)], dim: usize, ) -> Result<(Self::Storage, Shape)>

fn view_dispatch( src: &Self::Storage, src_l: &Layout, dst_l: &Layout, view: ViewOp, ) -> Result<Option<Self::Storage>>

impl Storage<Cpu, Int> for CpuIntStorage

fn dtype(&self) -> IntDType

fn device(&self) -> &Cpu

impl<D: Device, D2: Device> TransferDTypeKind<D, D2> for Int

fn transfer(src: &Tensor<D, Int>, device: &D2) -> Result<Tensor<D2, Int>>