Searched defs:diag_embed (Results 1 – 3 of 3) sorted by relevance
495 at::Tensor LazyNativeFunctions::diag_embed( in diag_embed() function in torch::lazy::LazyNativeFunctions
4341 def diag_embed( function
1122 Tensor diag_embed(const Tensor& self, int64_t offset, int64_t dim1_, int64_t dim2_) { in diag_embed() function