Searched defs:_remove_batch_dim (Results 1 – 2 of 2) sorted by relevance
106 Tensor _remove_batch_dim(const Tensor& self, int64_t level, int64_t batch_size, int64_t out_dim) { in _remove_batch_dim() function
143 Tensor _remove_batch_dim( in _remove_batch_dim() function