Function at::diagonal(const at::Tensor&, at::Dimname, at::Dimname, at::Dimname, int64_t)

• Defined in File Functions.h

Function Documentation

inline at::Tensor at::diagonal(const at::Tensor &self, at::Dimname outdim, at::Dimname dim1, at::Dimname dim2, int64_t offset = 0)