def make_node(self, input_):
defs_type = tensor.TensorType('int64', [False, False])
def_mask_type = tensor.TensorType('float32', [False, False])
# both type happened to be the same, but this is just a coincidence
def_map_type = defs_type
return theano.Apply(
self, [input_], [defs_type(), def_mask_type(), def_map_type()])
评论列表
文章目录