18 lines
221 B
Go
18 lines
221 B
Go
|
package my_nn
|
||
|
|
|
||
|
|
import (
|
||
|
|
torch "git.andr3h3nriqu3s.com/andr3/gotch/ts"
|
||
|
|
)
|
||
|
|
|
||
|
|
func or_panic(err error) {
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type MyLayer interface {
|
||
|
|
torch.ModuleT
|
||
|
|
|
||
|
|
ExtractFromVarstore(vs *VarStore)
|
||
|
|
}
|