search

wangkuiyi / gotorch

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

Add test cases to nn/initialization #305

Closed wangkuiyi closed 4 years ago

wangkuiyi commented 4 years ago

I changed Tensor.Index([]int64) into Tensor.Index[...int64).