MCPcopy Create free account
hub / github.com/ml-explore/mlx-data / squeeze

Function squeeze

mlx/data/Array.cpp:227–233  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

225}
226
227std::shared_ptr<Array> squeeze(
228 const std::shared_ptr<const Array>& arr,
229 const std::vector<int>& dims) {
230 auto dst = std::make_shared<Array>(arr);
231 dst->squeeze(dims);
232 return dst;
233}
234
235template <class T>
236void array_pad(

Callers 1

apply_keyMethod · 0.85

Calls 1

squeezeMethod · 0.45

Tested by

no test coverage detected