19 lines
233 B
Go
19 lines
233 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)
|
|
Debug()
|
|
}
|