def indexed(self, o, child, multiindex):
indices = multiindex.indices()
if isinstance(child, ListTensor) and all(isinstance(i, FixedIndex) for i in indices):
if len(indices) == 1:
return child.ufl_operands[indices[0]._value]
else:
return ListTensor(*(child.ufl_operands[i._value] for i in multiindex.indices()))
return self.expr(o, child, multiindex)
def list_tensor(self, o, *dops):
return ListTensor(*dops)