mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-23 14:59:34 +08:00
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:
committed by
Facebook Github Bot
parent
b2e94a7bcb
commit
8de1ce57d2
21
caffe2/python/helpers/algebra.py
Normal file
21
caffe2/python/helpers/algebra.py
Normal 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)
|
Reference in New Issue
Block a user