search

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.
BSD 3-Clause "New" or "Revised" License
11 stars 2 forks source link

Fix GatherLayer backward #170

Closed bricewang closed 4 months ago

bricewang commented 4 months ago

Add sum reduction to backward grad