Add Algebra and train helpers and proxy them to CNNMH

Summary: Add Algebra and train helpers and proxy them to CNNMH

Reviewed By: salexspb

Differential Revision: D4855040

fbshipit-source-id: d948ea913f674a6e47c4b72629a2d33253cb3130
This commit is contained in:
Yiming Wu
2017-04-11 23:00:27 -07:00
committed by Facebook Github Bot
parent b2e94a7bcb
commit 8de1ce57d2
4 changed files with 80 additions and 37 deletions

View File

@ -0,0 +1,21 @@
## @package algebra
# Module caffe2.python.helpers.algebra
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
__all__ = [
'Transpose',
'Sum',
]
def Transpose(model, blob_in, blob_out, **kwargs):
"""Transpose."""
return model.net.Transpose(blob_in, blob_out, **kwargs)
def Sum(model, blob_in, blob_out, **kwargs):
"""Sum"""
return model.net.Sum(blob_in, blob_out, **kwargs)