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)