burn::tensor

Trait TensorMetadata

pub trait TensorMetadata:
    Clone
    + Send
    + Sync
    + Debug {
    // Required methods
    fn dtype(&self) -> DType;
    fn shape(&self) -> Shape;
}
Expand description

Tensor metadata trait for tensor primitive.

Required Methods§

fn dtype(&self) -> DType

The dtype of the tensor.

fn shape(&self) -> Shape

The shape of the tensor.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types§

§

impl<C> TensorMetadata for RouterTensor<C>
where C: RunnerClient,

§

fn dtype(&self) -> DType

§

fn shape(&self) -> Shape

§

impl<R> TensorMetadata for FusionTensor<R>
where R: FusionRuntime,

§

fn dtype(&self) -> DType

§

fn shape(&self) -> Shape

Implementors§