original_scalar_value Class — pytorch Architecture

Source Code

aten/src/ATen/TensorIterator.h lines 393–405

  template <typename T>
  T original_scalar_value(int64_t arg) {
    auto& original_tensor_base = operands_[arg].original_tensor_base();
    if (original_tensor_base.defined()) {
      TORCH_INTERNAL_ASSERT(
          original_tensor_base.scalar_type() != common_dtype());
      return c10::fetch_and_cast<T>(
          original_tensor_base.scalar_type(),
          original_tensor_base.const_data_ptr());
    } else {
      return scalar_value<T>(arg);
    }
  }