Function PyScalarTensorCheck

oneflow/api/python/functional/common.cpp:174–181

Scalar Tensor

// Scalar Tensor
bool PyScalarTensorCheck(PyObject* obj) {
  if (!LazyMode::is_enabled() && PyTensor_Check(obj)) {
    const auto& tensor = PyTensor_Unpack(obj);
    return tensor->shape()->size() == 0
           && IsTriviallyCopyableDataType(tensor->dtype()->data_type());
  }
  return false;
}

Scalar PyUnpackScalarTensor(PyObject* obj) {
  if (PyBoolScalarTensorCheck(obj)) {