R/torch_eye_embed.R

Defines functions torch_eye_embed

#' @export
torch_eye_embed<-function(n_batch,n_dim)
{
  Id = torch_ones(n_batch,n_dim)
  torch_diag_embed(Id)
}
adsb85/lqp documentation built on April 9, 2022, 12:35 a.m.