Struct ITensorOperations

Derived Type

struct ITensorOperations

Subclassed by morpheus::ITensor

Public Functions

virtual std::shared_ptr<ITensor> slice(const std::vector<TensorIndex> &min_dims, const std::vector<TensorIndex> &max_dims) const = 0
virtual std::shared_ptr<ITensor> reshape(const std::vector<TensorIndex> &dims) const = 0
virtual std::shared_ptr<ITensor> deep_copy() const = 0
virtual std::shared_ptr<ITensor> copy_rows(const std::vector<std::pair<TensorIndex, TensorIndex>> &selected_rows, TensorIndex num_rows) const = 0
virtual std::shared_ptr<ITensor> as_type(DataType dtype) const = 0

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.