Trait burn_tensor::ops::TensorOps

source · 
pub trait TensorOps<B: Backend> {

Show 61 methods    fn from_data<const D: usize>(
        data: Data<B::Elem, D>,
        device: B::Device
    ) -> B::TensorPrimitive<D>;
    fn from_data_bool<const D: usize>(
        data: Data<bool, D>,
        device: B::Device
    ) -> B::BoolTensorPrimitive<D>;
    fn random<const D: usize>(
        shape: Shape<D>,
        distribution: Distribution<B::Elem>,
        device: B::Device
    ) -> B::TensorPrimitive<D>;
    fn shape<const D: usize>(tensor: &B::TensorPrimitive<D>) -> &Shape<D>;
    fn to_data<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> Data<B::Elem, D>;
    fn into_data<const D: usize>(
        tensor: B::TensorPrimitive<D>
    ) -> Data<B::Elem, D>;
    fn bool_shape<const D: usize>(
        tensor: &B::BoolTensorPrimitive<D>
    ) -> &Shape<D>;
    fn bool_to_data<const D: usize>(
        tensor: &B::BoolTensorPrimitive<D>
    ) -> Data<bool, D>;
    fn bool_into_data<const D: usize>(
        tensor: B::BoolTensorPrimitive<D>
    ) -> Data<bool, D>;
    fn bool_to_device<const D: usize>(
        tensor: &B::BoolTensorPrimitive<D>,
        device: B::Device
    ) -> B::BoolTensorPrimitive<D>;
    fn bool_reshape<const D1: usize, const D2: usize>(
        tensor: &B::BoolTensorPrimitive<D1>,
        shape: Shape<D2>
    ) -> B::BoolTensorPrimitive<D2>;
    fn bool_index<const D1: usize, const D2: usize>(
        tensor: &B::BoolTensorPrimitive<D1>,
        indexes: [Range<usize>; D2]
    ) -> B::BoolTensorPrimitive<D1>;
    fn device<const D: usize>(tensor: &B::TensorPrimitive<D>) -> B::Device;
    fn to_device<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        device: B::Device
    ) -> B::TensorPrimitive<D>;
    fn empty<const D: usize>(
        shape: Shape<D>,
        device: B::Device
    ) -> B::TensorPrimitive<D>;
    fn add<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn add_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::TensorPrimitive<D>;
    fn sub<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn sub_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::TensorPrimitive<D>;
    fn mul<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn mul_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::TensorPrimitive<D>;
    fn div<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn div_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::TensorPrimitive<D>;
    fn matmul<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn neg<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn swap_dims<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        dim1: usize,
        dim2: usize
    ) -> B::TensorPrimitive<D>;
    fn reshape<const D1: usize, const D2: usize>(
        tensor: &B::TensorPrimitive<D1>,
        shape: Shape<D2>
    ) -> B::TensorPrimitive<D2>;
    fn index<const D1: usize, const D2: usize>(
        tensor: &B::TensorPrimitive<D1>,
        indexes: [Range<usize>; D2]
    ) -> B::TensorPrimitive<D1>;
    fn index_assign<const D1: usize, const D2: usize>(
        tensor: &B::TensorPrimitive<D1>,
        indexes: [Range<usize>; D2],
        value: &B::TensorPrimitive<D1>
    ) -> B::TensorPrimitive<D1>;
    fn mask_fill<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        mask: &B::BoolTensorPrimitive<D>,
        value: B::Elem
    ) -> B::TensorPrimitive<D>;
    fn equal<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::BoolTensorPrimitive<D>;
    fn equal_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::BoolTensorPrimitive<D>;
    fn greater<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::BoolTensorPrimitive<D>;
    fn greater_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::BoolTensorPrimitive<D>;
    fn greater_equal<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::BoolTensorPrimitive<D>;
    fn greater_equal_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::BoolTensorPrimitive<D>;
    fn lower<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::BoolTensorPrimitive<D>;
    fn lower_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::BoolTensorPrimitive<D>;
    fn lower_equal<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::TensorPrimitive<D>
    ) -> B::BoolTensorPrimitive<D>;
    fn lower_equal_scalar<const D: usize>(
        lhs: &B::TensorPrimitive<D>,
        rhs: &B::Elem
    ) -> B::BoolTensorPrimitive<D>;
    fn detach<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn mean<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<1>;
    fn sum<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<1>;
    fn mean_dim<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        dim: usize
    ) -> B::TensorPrimitive<D>;
    fn sum_dim<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        dim: usize
    ) -> B::TensorPrimitive<D>;
    fn to_full_precision<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> <B::FullPrecisionBackend as Backend>::TensorPrimitive<D>;
    fn from_full_precision<const D: usize>(
        tensor: &<B::FullPrecisionBackend as Backend>::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn argmax<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        dim: usize
    ) -> <B::IntegerBackend as Backend>::TensorPrimitive<D>;
    fn argmin<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        dim: usize
    ) -> <B::IntegerBackend as Backend>::TensorPrimitive<D>;
    fn exp<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn log<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn powf<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        value: f32
    ) -> B::TensorPrimitive<D>;
    fn sqrt<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn erf<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;
    fn cat<const D: usize>(
        tensors: &[B::TensorPrimitive<D>],
        dim: usize
    ) -> B::TensorPrimitive<D>;
    fn relu<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D>;

    fn zeros<const D: usize>(
        shape: Shape<D>,
        device: B::Device
    ) -> B::TensorPrimitive<D> { ... }
    fn ones<const D: usize>(
        shape: Shape<D>,
        device: B::Device
    ) -> B::TensorPrimitive<D> { ... }
    fn arange(
        range: Range<usize>,
        device: B::Device
    ) -> <B::IntegerBackend as Backend>::TensorPrimitive<1> { ... }
    fn repeat<const D: usize>(
        tensor: &B::TensorPrimitive<D>,
        dim: usize,
        times: usize
    ) -> B::TensorPrimitive<D> { ... }
    fn transpose<const D: usize>(
        tensor: &B::TensorPrimitive<D>
    ) -> B::TensorPrimitive<D> { ... }

}

Required Methods§

Provided Methods§

Implementors§