Tanh.py 381 Bytes
Newer Older
yaoht's avatar
yaoht committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#coding:utf-8
import src.c2oObject as Node


def getOutShape(input_shape):
    # 获取output_shape
    return input_shape

# 构建节点


def createTanh(layer, nodename, inname, outname, input_shape):
    output_shape = getOutShape(input_shape)
    node = Node.c2oNode(layer, nodename, "Tanh", inname,
                        outname, input_shape, output_shape)
    return node