Method getExtraTensorViewInTensor

mllm/core/Tensor.cpp:113–118

}

Tensor Tensor::getExtraTensorViewInTensor(const std::string& extra_tensor_name) {
  if (impl_->attachedViews().count(extra_tensor_name) != 1) {
    MLLM_ERROR_EXIT(ExitCode::kCoreError, "Can't find {}", extra_tensor_name);
  }
  return Tensor(impl_->attachedViews().at(extra_tensor_name).second);
}

Tensor Tensor::zeros(const std::vector<int32_t>& shape, DataTypes dtype, DeviceTypes device) {
  auto i = Tensor::empty(shape, dtype, device).alloc();