Struct ITensor

(Latest Version)

Base Types

Derived Type

struct ITensor : public morpheus::ITensorStorage, public morpheus::ITensorOperations

Subclassed by morpheus::RMMTensor

Public Functions

~ITensor() override = default

virtual RankType rank() const = 0

virtual TensorIndex count() const = 0

virtual DType dtype() const = 0

virtual TensorIndex shape(TensorIndex) const = 0

virtual TensorIndex stride(TensorIndex) const = 0

virtual intptr_t stream() const = 0

virtual bool is_compact() const = 0

inline ShapeType get_shape() const

inline ShapeType get_stride() const

© Copyright 2023, NVIDIA. Last updated on Apr 11, 2023.