|
|
|
@ -465,6 +465,7 @@ std::string Tensor::GetShapeAndDataTypeInfo() const {
|
|
|
|
|
std::string Tensor::ToString() const {
|
|
|
|
|
const int small_tensor_size = 30;
|
|
|
|
|
std::ostringstream buf;
|
|
|
|
|
data_sync();
|
|
|
|
|
buf << "Tensor shape:[" << shape() << "]" << this->Dtype()->ToString();
|
|
|
|
|
// only print small tensor
|
|
|
|
|
if (DataSize() < small_tensor_size) {
|
|
|
|
@ -477,6 +478,7 @@ std::string Tensor::ToStringRepr() const {
|
|
|
|
|
std::ostringstream buf;
|
|
|
|
|
auto type_ptr = this->Dtype();
|
|
|
|
|
MS_EXCEPTION_IF_NULL(type_ptr);
|
|
|
|
|
data_sync();
|
|
|
|
|
buf << "Tensor shape:[" << shape() << "]" << type_ptr->ToString();
|
|
|
|
|
buf << "\nvalue:" << data().ToString(data_type_, shape());
|
|
|
|
|
return buf.str();
|
|
|
|
|