search

wangkuiyi / gotorch

A Go idiomatic binding to the C++ core of PyTorch
MIT License
316 stars 35 forks source link

Fix 0-dim shape calculation #291

Closed lhw362950217 closed 4 years ago

lhw362950217 commented 4 years ago

Polish Shape function to support 0-dim tensor.