Searched defs:diagonal_scatter (Results 1 – 3 of 3) sorted by relevance
4279 def diagonal_scatter( function
1586 def diagonal_scatter(input, src, offset: int = 0, dim1: int = 0, dim2: int = 1): function
4017 at::Tensor diagonal_scatter(const at::Tensor& self, const at::Tensor& src, int64_t offset, int64_t … in diagonal_scatter() function