caldera.utils.scatter_coo

caldera.utils.scatter_coo(indices, source, size=None, expand=False, dtype=None)

Scatter the provided source tensor to the provided indices.

Parameters

• indices (LongTensor) –

• source (FloatTensor) –

• size (Union[Size, Tuple[int, …], None]) –

• expand (bool) –

• dtype (Optional[Type]) –

Return type

FloatTensor

Returns