Trait burn_tensor::ops::TensorOps
source · pub trait TensorOps<B: Backend> {
Show 65 methods
// Required methods
fn from_data<const D: usize>(
data: Data<B::FloatElem, D>,
device: &B::Device
) -> B::TensorPrimitive<D>;
fn random<const D: usize>(
shape: Shape<D>,
distribution: Distribution<B::FloatElem>,
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::FloatElem, D>;
fn into_data<const D: usize>(
tensor: B::TensorPrimitive<D>
) -> Data<B::FloatElem, D>;
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::FloatElem
) -> 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::FloatElem
) -> 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::FloatElem
) -> 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::FloatElem
) -> 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_select<const D: usize>(
tensor: B::TensorPrimitive<D>,
indexes: B::IntTensorPrimitive<D>
) -> B::TensorPrimitive<D>;
fn index_select_assign<const D: usize>(
tensor: B::TensorPrimitive<D>,
indexes: B::IntTensorPrimitive<D>,
value: B::TensorPrimitive<D>
) -> B::TensorPrimitive<D>;
fn index_select_dim<const D: usize>(
tensor: B::TensorPrimitive<D>,
dim: usize,
indexes: B::IntTensorPrimitive<1>
) -> B::TensorPrimitive<D>;
fn index_select_dim_assign<const D1: usize, const D2: usize>(
tensor: B::TensorPrimitive<D1>,
dim: usize,
indexes: B::IntTensorPrimitive<1>,
value: B::TensorPrimitive<D2>
) -> B::TensorPrimitive<D1>;
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_scatter<const D: usize>(
tensor: B::TensorPrimitive<D>,
mask: B::BoolTensorPrimitive<D>,
source: B::TensorPrimitive<D>
) -> B::TensorPrimitive<D>;
fn mask_fill<const D: usize>(
tensor: B::TensorPrimitive<D>,
mask: B::BoolTensorPrimitive<D>,
value: B::FloatElem
) -> B::TensorPrimitive<D>;
fn equal<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::TensorPrimitive<D>
) -> B::BoolTensorPrimitive<D>;
fn equal_elem<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::FloatElem
) -> B::BoolTensorPrimitive<D>;
fn greater<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::TensorPrimitive<D>
) -> B::BoolTensorPrimitive<D>;
fn greater_elem<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::FloatElem
) -> B::BoolTensorPrimitive<D>;
fn greater_equal<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::TensorPrimitive<D>
) -> B::BoolTensorPrimitive<D>;
fn greater_equal_elem<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::FloatElem
) -> B::BoolTensorPrimitive<D>;
fn lower<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::TensorPrimitive<D>
) -> B::BoolTensorPrimitive<D>;
fn lower_elem<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::FloatElem
) -> B::BoolTensorPrimitive<D>;
fn lower_equal<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::TensorPrimitive<D>
) -> B::BoolTensorPrimitive<D>;
fn lower_equal_elem<const D: usize>(
lhs: B::TensorPrimitive<D>,
rhs: B::FloatElem
) -> B::BoolTensorPrimitive<D>;
fn sum<const D: usize>(
tensor: B::TensorPrimitive<D>
) -> B::TensorPrimitive<1>;
fn sum_dim<const D: usize>(
tensor: B::TensorPrimitive<D>,
dim: usize
) -> B::TensorPrimitive<D>;
fn mean<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 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::IntTensorPrimitive<D>;
fn argmin<const D: usize>(
tensor: B::TensorPrimitive<D>,
dim: usize
) -> B::IntTensorPrimitive<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 log1p<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 cos<const D: usize>(
tensor: B::TensorPrimitive<D>
) -> B::TensorPrimitive<D>;
fn sin<const D: usize>(
tensor: B::TensorPrimitive<D>
) -> B::TensorPrimitive<D>;
fn tanh<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: Vec<B::TensorPrimitive<D>>,
dim: usize
) -> B::TensorPrimitive<D>;
fn relu<const D: usize>(
tensor: B::TensorPrimitive<D>
) -> B::TensorPrimitive<D>;
// Provided methods
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::IntTensorPrimitive<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> { ... }
fn detach<const D: usize>(
tensor: B::TensorPrimitive<D>
) -> B::TensorPrimitive<D> { ... }
fn set_require_grad<const D: usize>(
tensor: B::TensorPrimitive<D>,
_require_grad: bool
) -> B::TensorPrimitive<D> { ... }
fn is_require_grad<const D: usize>(_tensor: &B::TensorPrimitive<D>) -> bool { ... }
}
Expand description
Operations on float tensors.