Ejemplos de indice_conv_backward en Python

Ejemplo n.º 1

Archivo: functional.py Proyecto: chodi150/scanrefer-pointgroup

    def backward(ctx, grad_output):
        indice_pairs, indice_pair_num, features, filters = ctx.saved_tensors
        input_bp, filters_bp = ops.indice_conv_backward(
            features, filters, grad_output.contiguous(), indice_pairs,
            indice_pair_num, False, True)

        return input_bp, filters_bp, None, None, None

Ejemplo n.º 2

    def backward(ctx, grad_output):
        indice_pairs, indice_pair_num, features, filters = ctx.saved_tensors
        input_bp, filters_bp = ops.indice_conv_backward(features,
                                                        filters,
                                                        grad_output,
                                                        indice_pairs,
                                                        indice_pair_num,
                                                        True,
                                                        False,
                                                        algo=ctx.algo)

        return input_bp, filters_bp, None, None, None, None