`*ts.CModule` does not implement `ts.Module` interface correctly
GoosvandenBekerom opened this issue · 0 comments
GoosvandenBekerom commented
This GoDoc comment
Line 1188 in f45f0a7
Implies that
*ts.CModule
implements the ts.Module
interface.
type Module interface {
// ModuleT
Forward(xs *Tensor) *Tensor
}
But since it returns an error, it does not, not sure if this is an issue, but just something I noticed while trying to use CModule as a Module.
// Forwad implements Module interface for CModule.
func (cm *CModule) Forward(tensor *Tensor) (*Tensor, error) {
var tensors []*Tensor = []*Tensor{tensor}
return cm.ForwardTs(tensors)
}