Trait tch_utils::tensor_ext::TensorExt

pub trait TensorExt {
    // Required methods
    fn sum_dim(&self, dim: i64) -> Tensor;
    fn sum_kdim(&self, dim: i64) -> Tensor;
    fn sum_dims<const D: usize>(&self, dims: [i64; D]) -> Tensor;
    fn sum_kdims<const D: usize>(&self, dims: [i64; D]) -> Tensor;
}

Required Methods

fn sum_dim(&self, dim: i64) -> Tensor

fn sum_kdim(&self, dim: i64) -> Tensor

fn sum_dims<const D: usize>(&self, dims: [i64; D]) -> Tensor

fn sum_kdims<const D: usize>(&self, dims: [i64; D]) -> Tensor

Implementations on Foreign Types

impl TensorExt for Tensor

fn sum_dim(&self, dim: i64) -> Tensor

fn sum_kdim(&self, dim: i64) -> Tensor

fn sum_dims<const D: usize>(&self, dims: [i64; D]) -> Tensor

fn sum_kdims<const D: usize>(&self, dims: [i64; D]) -> Tensor

Implementors