提交 937135df authored 作者: Razvan Pascanu's avatar Razvan Pascanu

R operator for AdvancedSubtensor

上级 94bdcd68
......@@ -4928,6 +4928,11 @@ class AdvancedSubtensor(Op):
raise NotImplementedError('Advanced indexing of x with arguments (%s) not supported yet'\
% ','.join(str(input) for input in inputs))
def R_op(self, inputs, eval_points):
if eval_points[0] is None:
return [None]
return self.make_node(eval_points[0], *inputs[1:]).outputs
def infer_shape(self, node, ishapes):
# Really special case
if len(ishapes) == 3:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论