Block_reduce with torch tensors on GPU

后端 未结 0 617
庸人自扰
庸人自扰 2021-02-05 06:31

I have a 2D tensor and I am trying to reduce it by taking average of every 4 columns. For numpy arrays this can be done by using the block_reduce function from skimage.measure m

相关标签:
回答
  • 消灭零回复
提交回复
热议问题