Struct ITensorOperations

(Latest Version)
struct ITensorOperations

Subclassed by morpheus::ITensor

Public Functions

virtual std::shared_ptr<ITensor> slice(const ShapeType &min_dims, const ShapeType &max_dims) const = 0

virtual std::shared_ptr<ITensor> reshape(const ShapeType &dims) const = 0

virtual std::shared_ptr<ITensor> deep_copy() const = 0

virtual std::shared_ptr<ITensor> copy_rows(const std::vector<RangeType> &selected_rows, TensorIndex num_rows) const = 0

virtual std::shared_ptr<ITensor> as_type(DType dtype) const = 0

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