Python indices_from_subtensor Examples

Programming Language: Python

Namespace/Package Name: aesara.tensor.subtensor

Method/Function: indices_from_subtensor

Examples at hotexamples.com: 2

Python indices_from_subtensor - 2 examples found. These are the top rated real world Python examples of aesara.tensor.subtensor.indices_from_subtensor extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

    def subtensor(x, *ilists):

        indices = indices_from_subtensor(ilists, idx_list)

        if len(indices) == 1:
            indices = indices[0]

        return x.__getitem__(indices)

Example #2

Show file

    def incsubtensor(x, y, *ilist, jax_fn=jax_fn, idx_list=idx_list):
        indices = indices_from_subtensor(ilist, idx_list)
        if len(indices) == 1:
            indices = indices[0]

        return jax_fn(x, indices, y)