pub struct Int;
Expand description
A type-level representation of the kind of a int tensor.
Trait Implementations§
source§impl<B> BasicAutodiffOps<B> for Intwhere
B: AutodiffBackend,
impl<B> BasicAutodiffOps<B> for Intwhere
B: AutodiffBackend,
source§fn inner<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <<Int as BasicAutodiffOps<B>>::InnerKind as TensorKind<<B as AutodiffBackend>::InnerBackend>>::Primitive<D>
fn inner<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <<Int as BasicAutodiffOps<B>>::InnerKind as TensorKind<<B as AutodiffBackend>::InnerBackend>>::Primitive<D>
Returns the inner tensor without the autodiff information. Read more
source§fn from_inner<const D: usize>(
inner: <<Int as BasicAutodiffOps<B>>::InnerKind as TensorKind<<B as AutodiffBackend>::InnerBackend>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn from_inner<const D: usize>( inner: <<Int as BasicAutodiffOps<B>>::InnerKind as TensorKind<<B as AutodiffBackend>::InnerBackend>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Convert a tensor to the autodiff backend. Read more
source§impl<B> BasicOps<B> for Intwhere
B: Backend,
impl<B> BasicOps<B> for Intwhere
B: Backend,
source§fn empty<const D: usize>(
shape: Shape<D>,
device: &<B as Backend>::Device
) -> <Int as TensorKind<B>>::Primitive<D>
fn empty<const D: usize>( shape: Shape<D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>
Creates an empty tensor with the given shape. Read more
source§fn shape<const D: usize>(
tensor: &<Int as TensorKind<B>>::Primitive<D>
) -> Shape<D>
fn shape<const D: usize>( tensor: &<Int as TensorKind<B>>::Primitive<D> ) -> Shape<D>
Returns the shape of the tensor. Read more
source§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 reshape<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, shape: Shape<D2> ) -> <Int as TensorKind<B>>::Primitive<D2>
Reshapes the tensor. Read more
source§fn transpose<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn transpose<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Transposes a tensor. Read more
source§fn swap_dims<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim1: usize,
dim2: usize
) -> <Int as TensorKind<B>>::Primitive<D>
fn swap_dims<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim1: usize, dim2: usize ) -> <Int as TensorKind<B>>::Primitive<D>
Swaps two dimensions of a tensor. Read more
source§fn slice<const D1: usize, const D2: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D1>,
ranges: [Range<usize>; D2]
) -> <Int as TensorKind<B>>::Primitive<D1>
fn slice<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, ranges: [Range<usize>; D2] ) -> <Int as TensorKind<B>>::Primitive<D1>
Select tensor elements corresponding for the given ranges. Read more
source§fn slice_assign<const D1: usize, const D2: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D1>,
ranges: [Range<usize>; D2],
value: <Int as TensorKind<B>>::Primitive<D1>
) -> <Int as TensorKind<B>>::Primitive<D1>
fn slice_assign<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, ranges: [Range<usize>; D2], value: <Int as TensorKind<B>>::Primitive<D1> ) -> <Int as TensorKind<B>>::Primitive<D1>
Assigns the given value to the tensor elements corresponding for the given ranges. Read more
source§fn device<const D: usize>(
tensor: &<Int as TensorKind<B>>::Primitive<D>
) -> <B as Backend>::Device
fn device<const D: usize>( tensor: &<Int as TensorKind<B>>::Primitive<D> ) -> <B as Backend>::Device
Returns the device on which the tensor is allocated. Read more
source§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 to_device<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>
Moves the tensor to the given device. Read more
source§fn into_data<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> Reader<Data<<Int as BasicOps<B>>::Elem, D>>
fn into_data<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> Reader<Data<<Int as BasicOps<B>>::Elem, D>>
Extracts the data from the tensor. Read more
source§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 from_data<const D: usize>( data: Data<<Int as BasicOps<B>>::Elem, D>, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>
Creates a tensor from the given data. Read more
source§fn repeat<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize,
times: usize
) -> <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>
Repeat the tensor along the given dimension. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, D, Bool>
Equates the given tensors. Read more
source§fn not_equal<const D: usize>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: <Int as TensorKind<B>>::Primitive<D>
) -> Tensor<B, D, Bool>
fn not_equal<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, D, Bool>
Applies element-wise non-equality comparison between the given tensors. Read more
source§fn cat<const D: usize>(
vectors: Vec<<Int as TensorKind<B>>::Primitive<D>>,
dim: usize
) -> <Int as TensorKind<B>>::Primitive<D>
fn cat<const D: usize>( vectors: Vec<<Int as TensorKind<B>>::Primitive<D>>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>
Concatenates the given tensors along the given dimension. Read more
source§fn any<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> Tensor<B, 1, Bool>
fn any<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, 1, Bool>
Tests if any element in the
tensor
evaluates to True. Read moresource§fn any_dim<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> Tensor<B, D, Bool>
fn any_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> Tensor<B, D, Bool>
Tests if any element in the tensor evaluates to True along a given dimension dim. Read more
source§fn all<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> Tensor<B, 1, Bool>
fn all<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, 1, Bool>
Tests if all elements in the
tensor
evaluate to True. Read moresource§fn all_dim<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> Tensor<B, D, Bool>
fn all_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> Tensor<B, D, Bool>
source§fn permute<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
axes: [usize; D]
) -> <Int as TensorKind<B>>::Primitive<D>
fn permute<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, axes: [usize; D] ) -> <Int as TensorKind<B>>::Primitive<D>
Permutes the dimensions of a tensor. Read more
source§fn expand<const D1: usize, const D2: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D1>,
shape: Shape<D2>
) -> <Int as TensorKind<B>>::Primitive<D2>
fn expand<const D1: usize, const D2: usize>( tensor: <Int as TensorKind<B>>::Primitive<D1>, shape: Shape<D2> ) -> <Int as TensorKind<B>>::Primitive<D2>
Broadcasts the given tensor to the specified shape. Read more
source§fn flip<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
axes: &[usize]
) -> <Int as TensorKind<B>>::Primitive<D>
fn flip<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, axes: &[usize] ) -> <Int as TensorKind<B>>::Primitive<D>
Flips the tensor along the given axes. Read more
source§fn elem_type_name() -> &'static str
fn elem_type_name() -> &'static str
Returns the name of the element type.
source§impl<B> Numeric<B> for Intwhere
B: Backend,
impl<B> Numeric<B> for Intwhere
B: Backend,
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Adds two tensors together. Read more
source§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 add_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
Adds a scalar to a tensor element-wise. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Subtracts two tensors. Read more
source§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 sub_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
Subtracts a scalar from a tensor element-wise. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Divides two tensors. Read more
source§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 div_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
Divides a tensor by a scalar element-wise. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Multiplies two tensors. Read more
source§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 mul_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
Multiplies a tensor by a scalar element-wise. Read more
source§fn neg<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn neg<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Negates a tensor. Read more
source§fn zeros<const D: usize>(
shape: Shape<D>,
device: &<B as Backend>::Device
) -> <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>
Creates a tensor filled with zeros. Read more
source§fn ones<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>
Creates a tensor filled with ones. Read more
source§fn full<const D: usize, E>(
shape: Shape<D>,
fill_value: E,
device: &<B as Backend>::Device
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
fn full<const D: usize, E>(
shape: Shape<D>,
fill_value: E,
device: &<B as Backend>::Device
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
Creates a tensor filled with elements equal to the given value. Read more
source§fn sum<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<1>
fn sum<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<1>
Sums all the elements of the tensor. Read more
source§fn sum_dim<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> <Int as TensorKind<B>>::Primitive<D>
fn sum_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>
Sums all the elements of the tensor along a dimension. Read more
source§fn prod<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<1>
fn prod<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<1>
Computes the product of all the elements of the tensor. Read more
source§fn prod_dim<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> <Int as TensorKind<B>>::Primitive<D>
fn prod_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>
Computes the product of all the elements of the tensor along a dimension. Read more
source§fn mean<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<1>
fn mean<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<1>
Computes the mean of all the elements of the tensor. Read more
source§fn mean_dim<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> <Int as TensorKind<B>>::Primitive<D>
fn mean_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>
Computes the mean of all the elements of the tensor along a dimension. Read more
source§fn equal_elem<const D: usize>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: <Int as BasicOps<B>>::Elem
) -> 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>
Element-wise equality between two tensors. Read more
source§fn not_equal_elem<const D: usize>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: <Int as BasicOps<B>>::Elem
) -> Tensor<B, D, Bool>
fn not_equal_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>
Element-wise non-equality between two tensors. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, D, Bool>
Element-wise greater than comparison between two tensors. Read more
source§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_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>
Element-wise greater than comparison between a tensor and a scalar. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, D, Bool>
Element-wise greater than or equal comparison between two tensors. Read more
source§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 greater_equal_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>
Element-wise greater than or equal comparison between a tensor and a scalar. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, D, Bool>
Element-wise less than comparison between two tensors. Read more
source§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_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>
Element-wise less than comparison between a tensor and a scalar. Read more
source§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<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> Tensor<B, D, Bool>
Element-wise less than or equal comparison between two tensors. Read more
source§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 lower_equal_elem<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as BasicOps<B>>::Elem ) -> Tensor<B, D, Bool>
Element-wise less than or equal comparison between a tensor and a scalar. Read more
source§fn mask_where<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_where<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>
Selects elements from a tensor based on a boolean mask. Read more
source§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 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>
Fills elements of a tensor based on a boolean mask. Read more
source§fn select<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize,
indices: Tensor<B, 1, Int>
) -> <Int as TensorKind<B>>::Primitive<D>
fn select<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize, indices: Tensor<B, 1, Int> ) -> <Int as TensorKind<B>>::Primitive<D>
Select tensor elements along the given dimension corresponding for the given indices. Read more
source§fn select_assign<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize,
indices: Tensor<B, 1, Int>,
values: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn select_assign<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize, indices: Tensor<B, 1, Int>, values: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Assign the selected elements along the given dimension corresponding to the given indices
from the value tensor. Read more
source§fn gather<const D: usize>(
dim: usize,
tensor: <Int as TensorKind<B>>::Primitive<D>,
indices: Tensor<B, D, Int>
) -> <Int as TensorKind<B>>::Primitive<D>
fn gather<const D: usize>( dim: usize, tensor: <Int as TensorKind<B>>::Primitive<D>, indices: Tensor<B, D, Int> ) -> <Int as TensorKind<B>>::Primitive<D>
Gathers elements from a tensor along an axis. Read more
source§fn scatter<const D: usize>(
dim: usize,
tensor: <Int as TensorKind<B>>::Primitive<D>,
indices: Tensor<B, D, Int>,
values: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn scatter<const D: usize>( dim: usize, tensor: <Int as TensorKind<B>>::Primitive<D>, indices: Tensor<B, D, Int>, values: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Scatters elements into a tensor along an axis. Read more
source§fn argmax<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> <B as Backend>::IntTensorPrimitive<D>
fn argmax<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <B as Backend>::IntTensorPrimitive<D>
Gets the indices of the maximum elements of a tensor along an axis. Read more
source§fn argmin<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> <B as Backend>::IntTensorPrimitive<D>
fn argmin<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <B as Backend>::IntTensorPrimitive<D>
Gets the indices of the minimum elements of a tensor along an axis. Read more
source§fn max<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<1>
fn max<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<1>
Gets the maximum elements of a tensor along an axis. Read more
source§fn max_dim<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> <Int as TensorKind<B>>::Primitive<D>
fn max_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>
Gets the maximum elements of a tensor along an axis. Read more
source§fn max_dim_with_indices<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> (<Int as TensorKind<B>>::Primitive<D>, <B as Backend>::IntTensorPrimitive<D>)
fn max_dim_with_indices<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> (<Int as TensorKind<B>>::Primitive<D>, <B as Backend>::IntTensorPrimitive<D>)
Gets the maximum elements of a tensor along an axis. Read more
source§fn min<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<1>
fn min<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<1>
Gets the minimum elements of a tensor along an axis. Read more
source§fn min_dim<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> <Int as TensorKind<B>>::Primitive<D>
fn min_dim<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> <Int as TensorKind<B>>::Primitive<D>
Gets the minimum elements of a tensor along an axis. Read more
source§fn min_dim_with_indices<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize
) -> (<Int as TensorKind<B>>::Primitive<D>, <B as Backend>::IntTensorPrimitive<D>)
fn min_dim_with_indices<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize ) -> (<Int as TensorKind<B>>::Primitive<D>, <B as Backend>::IntTensorPrimitive<D>)
Gets the minimum elements and indices of a tensor along an axis. Read more
source§fn clamp<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
min: <B as Backend>::IntElem,
max: <B as Backend>::IntElem
) -> <Int as TensorKind<B>>::Primitive<D>
fn clamp<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, min: <B as Backend>::IntElem, max: <B as Backend>::IntElem ) -> <Int as TensorKind<B>>::Primitive<D>
Clamp the tensor between the given min and max values. Read more
source§fn clamp_min<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
min: <B as Backend>::IntElem
) -> <Int as TensorKind<B>>::Primitive<D>
fn clamp_min<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, min: <B as Backend>::IntElem ) -> <Int as TensorKind<B>>::Primitive<D>
Clamps a tensor under a minimum value. Read more
source§fn clamp_max<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
max: <B as Backend>::IntElem
) -> <Int as TensorKind<B>>::Primitive<D>
fn clamp_max<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, max: <B as Backend>::IntElem ) -> <Int as TensorKind<B>>::Primitive<D>
Clamps a tensor over a maximum value. Read more
source§fn abs<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn abs<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Calculate absolute value on all elements of a tensor Read more
source§fn powf<const D: usize>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn powf<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Element-wise power of a tensor to a float tensor Read more
source§fn powf_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
fn powf_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
Element-wise power of a tensor to a scalar float Read more
source§fn powi<const D: usize>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn powi<const D: usize>( lhs: <Int as TensorKind<B>>::Primitive<D>, rhs: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Element-wise power of a tensor Read more
source§fn powi_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
fn powi_scalar<const D: usize, E>(
lhs: <Int as TensorKind<B>>::Primitive<D>,
rhs: E
) -> <Int as TensorKind<B>>::Primitive<D>where
E: ElementConversion,
Element-wise power of a tensor to a scalar int Read more
source§fn random<const D: usize>(
shape: Shape<D>,
distribution: Distribution,
device: &<B as Backend>::Device
) -> <Int as TensorKind<B>>::Primitive<D>
fn random<const D: usize>( shape: Shape<D>, distribution: Distribution, device: &<B as Backend>::Device ) -> <Int as TensorKind<B>>::Primitive<D>
Create a random tensor. Read more
source§fn sign<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>
) -> <Int as TensorKind<B>>::Primitive<D>
fn sign<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D> ) -> <Int as TensorKind<B>>::Primitive<D>
Returns the signs of the elements of a tensor. Read more
source§fn sort<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize,
descending: bool
) -> <Int as TensorKind<B>>::Primitive<D>
fn sort<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize, descending: bool ) -> <Int as TensorKind<B>>::Primitive<D>
Sort the elements of the input
tensor
by value along a given dimension. Read moresource§fn sort_with_indices<const D: usize>(
tensor: <Int as TensorKind<B>>::Primitive<D>,
dim: usize,
descending: bool
) -> (<Int as TensorKind<B>>::Primitive<D>, <Int as TensorKind<B>>::Primitive<D>)
fn sort_with_indices<const D: usize>( tensor: <Int as TensorKind<B>>::Primitive<D>, dim: usize, descending: bool ) -> (<Int as TensorKind<B>>::Primitive<D>, <Int as TensorKind<B>>::Primitive<D>)
Sort the elements of the input
tensor
by value along a given dimension. Read moreAuto Trait Implementations§
impl Freeze for Int
impl RefUnwindSafe for Int
impl Send for Int
impl Sync for Int
impl Unpin for Int
impl UnwindSafe for Int
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more