Trait burn::tensor::ops::IntTensorOps

pub trait IntTensorOps<B>where
    B: Backend,{
    // Required methods
    fn int_empty<const D: usize>(
        shape: Shape<D>,
        device: &<B as Backend>::Device
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_shape<const D: usize>(
        tensor: &<B as Backend>::IntTensorPrimitive<D>
    ) -> Shape<D>;
    fn int_into_data<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> Data<<B as Backend>::IntElem, D>;
    fn int_from_data<const D: usize>(
        data: Data<<B as Backend>::IntElem, D>,
        device: &<B as Backend>::Device
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_device<const D: usize>(
        tensor: &<B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::Device;
    fn int_to_device<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        device: &<B as Backend>::Device
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_reshape<const D1: usize, const D2: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D1>,
        shape: Shape<D2>
    ) -> <B as Backend>::IntTensorPrimitive<D2>;
    fn int_slice<const D1: usize, const D2: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D1>,
        indices: [Range<usize>; D2]
    ) -> <B as Backend>::IntTensorPrimitive<D1>;
    fn int_slice_assign<const D1: usize, const D2: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D1>,
        indices: [Range<usize>; D2],
        value: <B as Backend>::IntTensorPrimitive<D1>
    ) -> <B as Backend>::IntTensorPrimitive<D1>;
    fn int_into_float<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::TensorPrimitive<D>;
    fn int_mask_where<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        mask: <B as Backend>::BoolTensorPrimitive<D>,
        source: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_mask_fill<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        mask: <B as Backend>::BoolTensorPrimitive<D>,
        value: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_gather<const D: usize>(
        dim: usize,
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        indices: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_scatter<const D: usize>(
        dim: usize,
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        indices: <B as Backend>::IntTensorPrimitive<D>,
        value: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_select<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize,
        indices: <B as Backend>::IntTensorPrimitive<1>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_select_assign<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize,
        indices: <B as Backend>::IntTensorPrimitive<1>,
        value: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_cat<const D: usize>(
        tensors: Vec<<B as Backend>::IntTensorPrimitive<D>, Global>,
        dim: usize
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_equal<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_equal_elem<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_greater<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_greater_elem<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_greater_equal<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_greater_equal_elem<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_lower<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_lower_elem<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_lower_equal<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_lower_equal_elem<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::BoolTensorPrimitive<D>;
    fn int_add<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_add_scalar<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_sub<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_sub_scalar<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_mul<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_mul_scalar<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_div<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_div_scalar<const D: usize>(
        lhs: <B as Backend>::IntTensorPrimitive<D>,
        rhs: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_zeros<const D: usize>(
        shape: Shape<D>,
        device: &<B as Backend>::Device
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_ones<const D: usize>(
        shape: Shape<D>,
        device: &<B as Backend>::Device
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_sum<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<1>;
    fn int_sum_dim<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_mean_dim<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_argmax<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_argmin<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> <B as Backend>::IntTensorPrimitive<D>;
    fn int_abs<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D>;

    // Provided methods
    fn int_to_data<const D: usize>(
        tensor: &<B as Backend>::IntTensorPrimitive<D>
    ) -> Data<<B as Backend>::IntElem, D> { ... }
    fn int_repeat<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize,
        times: usize
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_clamp_min<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        min: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_clamp_max<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        max: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_clamp<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        min: <B as Backend>::IntElem,
        max: <B as Backend>::IntElem
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_neg<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_full<const D: usize>(
        shape: Shape<D>,
        fill_value: <B as Backend>::IntElem,
        device: &<B as Backend>::Device
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_mean<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<1> { ... }
    fn int_max<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<1> { ... }
    fn int_max_dim<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_max_dim_with_indices<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> (<B as Backend>::IntTensorPrimitive<D>, <B as Backend>::IntTensorPrimitive<D>) { ... }
    fn int_min<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>
    ) -> <B as Backend>::IntTensorPrimitive<1> { ... }
    fn int_min_dim<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> <B as Backend>::IntTensorPrimitive<D> { ... }
    fn int_min_dim_with_indices<const D: usize>(
        tensor: <B as Backend>::IntTensorPrimitive<D>,
        dim: usize
    ) -> (<B as Backend>::IntTensorPrimitive<D>, <B as Backend>::IntTensorPrimitive<D>) { ... }
}

Int Tensor API for basic and numeric operations, see tensor for documentation on each function.

Required Methods

fn int_empty<const D: usize>( shape: Shape<D>, device: &<B as Backend>::Device ) -> <B as Backend>::IntTensorPrimitive<D>

Creates a new int tensor.

Arguments

• shape - The shape of the tensor.
• device - The device to create the tensor on.

Returns

The integer tensor with the given shape.

fn int_shape<const D: usize>( tensor: &<B as Backend>::IntTensorPrimitive<D> ) -> Shape<D>

Returns the shape of the tensor.

Arguments

• tensor - The tensor.

Returns

The shape of the tensor.

fn int_into_data<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> Data<<B as Backend>::IntElem, D>

Converts the tensor to a data structure.

Arguments

• tensor - The tensor.

Returns

The data structure with the tensor’s data.

fn int_from_data<const D: usize>( data: Data<<B as Backend>::IntElem, D>, device: &<B as Backend>::Device ) -> <B as Backend>::IntTensorPrimitive<D>

Creates a tensor from the data structure.

Arguments

• data - The data structure.
• device - The device to create the tensor on.

Returns

The tensor with the data.

fn int_device<const D: usize>( tensor: &<B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::Device

Gets the device of the tensor.

Arguments

• tensor - The tensor.

Returns

The device of the tensor.

fn int_to_device<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, device: &<B as Backend>::Device ) -> <B as Backend>::IntTensorPrimitive<D>

Moves the tensor to the given device.

fn int_reshape<const D1: usize, const D2: usize>( tensor: <B as Backend>::IntTensorPrimitive<D1>, shape: Shape<D2> ) -> <B as Backend>::IntTensorPrimitive<D2>

Reshapes the tensor.

Arguments

• tensor - The tensor.
• shape - The new shape.

Returns

The tensor with the new shape.

fn int_slice<const D1: usize, const D2: usize>( tensor: <B as Backend>::IntTensorPrimitive<D1>, indices: [Range<usize>; D2] ) -> <B as Backend>::IntTensorPrimitive<D1>

Gets the element at the given indices.

Arguments

• tensor - The tensor.
• indices - The indices.

Returns

The elements at the given indices.

fn int_slice_assign<const D1: usize, const D2: usize>( tensor: <B as Backend>::IntTensorPrimitive<D1>, indices: [Range<usize>; D2], value: <B as Backend>::IntTensorPrimitive<D1> ) -> <B as Backend>::IntTensorPrimitive<D1>

Sets the element at the given indices.

Arguments

• tensor - The tensor.
• indices - The indices.

Returns

The tensor with the element at the given indices set.

fn int_into_float<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::TensorPrimitive<D>

Converts int tensor to float tensor.

Arguments

• tensor - The tensor.

Returns

The int tensor with the same data as the float tensor.

fn int_mask_where<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, mask: <B as Backend>::BoolTensorPrimitive<D>, source: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Fills the tensor with values from the source tensor if the mask is true at the given indices.

Arguments

• tensor - The tensor.
• mask - The mask.
• source - The source tensor.

Returns

The tensor with the values filled.

fn int_mask_fill<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, mask: <B as Backend>::BoolTensorPrimitive<D>, value: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Fills the tensor with the given value if the mask is true at the given indices.

Arguments

• tensor - The tensor.
• mask - The mask.
• value - The value.

Returns

The tensor with the values filled.

fn int_gather<const D: usize>( dim: usize, tensor: <B as Backend>::IntTensorPrimitive<D>, indices: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Gather elements from the tensor at the given indices.

Arguments

• dim - The dimension to gather from.
• tensor - The tensor.
• indices - The indices.

fn int_scatter<const D: usize>( dim: usize, tensor: <B as Backend>::IntTensorPrimitive<D>, indices: <B as Backend>::IntTensorPrimitive<D>, value: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Scatter a given value to the tensor at the given indices.

Arguments

• dim - The dimension to scatter to.
• tensor - The tensor.
• indices - The indices.
• value - The value.

Returns

The tensor with the values scattered.

fn int_select<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, dim: usize, indices: <B as Backend>::IntTensorPrimitive<1> ) -> <B as Backend>::IntTensorPrimitive<D>

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

Arguments

• tensor - The tensor.
• dim - The dimension to select from.
• indices - The indices.

Returns

The tensor with the selected elements.

fn int_select_assign<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, dim: usize, indices: <B as Backend>::IntTensorPrimitive<1>, value: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

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

Arguments

• tensor - The tensor.
• dim - The dimension to select from.
• indices - The indices.
• value - The value.

Returns

The tensor with the selected elements assigned to the given value.

fn int_cat<const D: usize>( tensors: Vec<<B as Backend>::IntTensorPrimitive<D>, Global>, dim: usize ) -> <B as Backend>::IntTensorPrimitive<D>

Concatenates the given tensors along the given dimension.

Arguments

• tensors - The tensors.
• dim - The dimension to concatenate along.

Returns

The concatenated tensor.

fn int_equal<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise equality comparison.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The boolean tensor with the result of the comparison.

fn int_equal_elem<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise equality comparison with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The boolean tensor with the result of the comparison.

fn int_greater<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise greater than comparison.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The boolean tensor with the result of the comparison.

fn int_greater_elem<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise greater than comparison with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The boolean tensor with the result of the comparison.

fn int_greater_equal<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise greater than or equal comparison.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The boolean tensor with the result of the comparison.

fn int_greater_equal_elem<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise greater than or equal comparison with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The boolean tensor with the result of the comparison.

fn int_lower<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise less than comparison.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The boolean tensor with the result of the comparison.

fn int_lower_elem<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise less than comparison with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The boolean tensor with the result of the comparison.

fn int_lower_equal<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise less than or equal comparison.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The boolean tensor with the result of the comparison.

fn int_lower_equal_elem<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::BoolTensorPrimitive<D>

Elementwise less than or equal comparison with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The boolean tensor with the result of the comparison.

fn int_add<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise addition.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The result of the addition.

fn int_add_scalar<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise addition with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The result of the addition.

fn int_sub<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise subtraction.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The result of the subtraction.

fn int_sub_scalar<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise subtraction with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The result of the subtraction.

fn int_mul<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise multiplication.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The result of the multiplication.

fn int_mul_scalar<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise multiplication with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The result of the multiplication.

fn int_div<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise division.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side tensor.

Returns

The result of the division.

fn int_div_scalar<const D: usize>( lhs: <B as Backend>::IntTensorPrimitive<D>, rhs: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise division with a scalar.

Arguments

• lhs - The left hand side tensor.
• rhs - The right hand side scalar.

Returns

The result of the division.

fn int_zeros<const D: usize>( shape: Shape<D>, device: &<B as Backend>::Device ) -> <B as Backend>::IntTensorPrimitive<D>

Creates a tensor of zeros.

Arguments

• shape - The shape of the tensor.
• device - The device to create the tensor on.

Returns

The tensor of zeros.

fn int_ones<const D: usize>( shape: Shape<D>, device: &<B as Backend>::Device ) -> <B as Backend>::IntTensorPrimitive<D>

Creates a tensor of ones.

Arguments

• shape - The shape of the tensor.
• device - The device to create the tensor on.

Returns

The tensor of ones.

fn int_sum<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<1>

Sums all elements in the tensor.

Arguments

• tensor - The tensor to sum.

Returns

The sum of all elements in the tensor.

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

Sums all elements in the tensor along a dimension.

Arguments

• tensor - The tensor to sum.
• dim - The dimension to sum along.

Returns

The sum of all elements in the tensor along the dimension.

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

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

Arguments

• tensor - The tensor to compute the mean of.

Returns

The mean of all elements in the tensor along the dimension.

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

Gets the indices of the maximum elements along a dimension.

Arguments

• tensor - The tensor to get the maximum indices of.
• dim - The dimension to get the maximum indices along.

Returns

The indices of the maximum elements along the dimension.

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

Gets the indices of the minimum elements along a dimension.

Arguments

• tensor - The tensor to get the minimum indices of.
• dim - The dimension to get the minimum indices along.

Returns

The indices of the minimum elements along the dimension.

fn int_abs<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Returns a new tensor with absolute values.

Arguments

• tensor - The tensor to take absolute value of.

Returns

A tensor with the same shape as tensor with absolute values.

Provided Methods

fn int_to_data<const D: usize>( tensor: &<B as Backend>::IntTensorPrimitive<D> ) -> Data<<B as Backend>::IntElem, D>

Gets the data from the tensor.

Arguments

• tensor - The tensor.

Returns

The data cloned from the data structure.

fn int_repeat<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, dim: usize, times: usize ) -> <B as Backend>::IntTensorPrimitive<D>

Repeats the tensor along the given dimension the given number of times.

Arguments

• tensor - The tensor.
• dim - The dimension to repeat.
• times - The number of times to repeat.

Returns

The tensor with the given dimension repeated the given number of times.

fn int_clamp_min<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, min: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Clamps a tensor under a minimum value.

Arguments

• tensor - The tensor to clamp.
• min - The minimum value.

Returns

The clamped tensor.

fn int_clamp_max<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, max: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Clamps a tensor over a maximum value.

Arguments

• tensor - The tensor to clamp.
• max - The maximum value.

Returns

The clamped tensor.

fn int_clamp<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D>, min: <B as Backend>::IntElem, max: <B as Backend>::IntElem ) -> <B as Backend>::IntTensorPrimitive<D>

Clamps a tensor between a minimum and maximum value.

Arguments

• tensor - The tensor to clamp.
• min - The minimum value.
• max - The maximum value.

Returns

The clamped tensor.

fn int_neg<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<D>

Elementwise negation.

Arguments

• tensor - The tensor to negate.

Returns

The negated tensor.

fn int_full<const D: usize>( shape: Shape<D>, fill_value: <B as Backend>::IntElem, device: &<B as Backend>::Device ) -> <B as Backend>::IntTensorPrimitive<D>

Creates a tensor filled with given value.

Arguments

• shape - The shape of the tensor.
• fill_value - The value with which to fill the tensor.
• device - The device to create the tensor on.

Returns

The tensor filled with given value

fn int_mean<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<1>

Computes the mean of all elements in the tensor.

Arguments

• tensor - The tensor to compute the mean of.

Returns

The mean of all elements in the tensor.

fn int_max<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<1>

Gets the maximum element in the tensor.

Arguments

• tensor - The tensor to get the maximum element of.

Returns

The maximum element in the tensor.

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

Gets the maximum element in the tensor along a dimension.

Arguments

• tensor - The tensor to get the maximum element of.
• dim - The dimension to get the maximum element along.

Returns

The maximum element in the tensor along the dimension.

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

Gets the maximum elements and corresponding indices along a dimension.

Arguments

• tensor - The tensor to get the maximum elements and indices of.
• dim - The dimension to get the maximum elements and indices along.

Returns

The maximum elements and corresponding indices along the dimension.

fn int_min<const D: usize>( tensor: <B as Backend>::IntTensorPrimitive<D> ) -> <B as Backend>::IntTensorPrimitive<1>

Gets the minimum element in the tensor.

Arguments

• tensor - The tensor to get the minimum element of.

Returns

The minimum element in the tensor.

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

Gets the minimum elements in the tensor along a dimension.

Arguments

• tensor - The tensor to get the minimum element of.
• dim - The dimension to get the minimum element along.

Returns

The minimum element in the tensor along the dimension.

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

Gets the minimum elements and corresponding indices along a dimension.

Arguments

• tensor - The tensor to get the minimum elements and indices of.
• dim - The dimension to get the minimum elements and indices along.

Returns

The minimum elements and corresponding indices along the dimension.

Implementors