簡單層的實現

class MulLayer: def __init__(self): self.x = None self.y = None def forward(self,x,y): self.x = x self.y = y out = x*y return out
相關文章
相關標籤/搜索