def create(self, input_shapes):
input_shapes = toMultiShape(input_shapes)
inputs = [
Variable(list(output_shape)) for output_shape in input_shapes
]
return LambdaLayer(input_vars=inputs,
out_var=self.function(*inputs),
input_shape=input_shapes)
def __init__(self, input_shape, node=None, jvalue=None, name=None):
if jvalue:
self.value = jvalue
self.bigdl_type = "float"
else:
if node:
super(Variable, self).__init__(jvalue, "float", node, name)
else:
super(Variable, self).__init__(jvalue, "float", toMultiShape(input_shape), name)
def __init__(self, input_shape, node=None, jvalue=None):
if jvalue:
self.value = jvalue
self.bigdl_type = "float"
else:
if node:
super(Variable, self).__init__(jvalue, "float", node)
else:
super(Variable, self).__init__(jvalue, "float",
toMultiShape(input_shape))
def create(self, input_shapes):
input_shapes = toMultiShape(input_shapes)
inputs = [Variable(list(output_shape)) for output_shape in input_shapes]
return LambdaLayer(input_vars=inputs,
out_var=self.function(*inputs),
input_shape=input_shapes)
