... dim=0, keepdim=True) h = torch.mul(o, F.tanh(c)) return c, h. 開發者ID:dasguptar,項目名稱:treelstm.pytorch,代碼行數:18,代碼來源:model.py ...
確定! 回上一頁