64 lines
1.8 KiB
JavaScript
64 lines
1.8 KiB
JavaScript
|
"use strict";
|
||
|
|
|
||
|
|
Object.defineProperty(exports, "__esModule", {
|
||
|
|
value: true
|
||
|
|
});
|
||
|
|
exports.createSqueeze = void 0;
|
||
|
|
|
||
|
|
var _object = require("../../utils/object.js");
|
||
|
|
|
||
|
|
var _array = require("../../utils/array.js");
|
||
|
|
|
||
|
|
var _factory = require("../../utils/factory.js");
|
||
|
|
|
||
|
|
var name = 'squeeze';
|
||
|
|
var dependencies = ['typed', 'matrix'];
|
||
|
|
var createSqueeze = /* #__PURE__ */(0, _factory.factory)(name, dependencies, function (_ref) {
|
||
|
|
var typed = _ref.typed,
|
||
|
|
matrix = _ref.matrix;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Squeeze a matrix, remove inner and outer singleton dimensions from a matrix.
|
||
|
|
*
|
||
|
|
* Syntax:
|
||
|
|
*
|
||
|
|
* math.squeeze(x)
|
||
|
|
*
|
||
|
|
* Examples:
|
||
|
|
*
|
||
|
|
* math.squeeze([3]) // returns 3
|
||
|
|
* math.squeeze([[3]]) // returns 3
|
||
|
|
*
|
||
|
|
* const A = math.zeros(3, 1) // returns [[0], [0], [0]] (size 3x1)
|
||
|
|
* math.squeeze(A) // returns [0, 0, 0] (size 3)
|
||
|
|
*
|
||
|
|
* const B = math.zeros(1, 3) // returns [[0, 0, 0]] (size 1x3)
|
||
|
|
* math.squeeze(B) // returns [0, 0, 0] (size 3)
|
||
|
|
*
|
||
|
|
* // only inner and outer dimensions are removed
|
||
|
|
* const C = math.zeros(2, 1, 3) // returns [[[0, 0, 0]], [[0, 0, 0]]] (size 2x1x3)
|
||
|
|
* math.squeeze(C) // returns [[[0, 0, 0]], [[0, 0, 0]]] (size 2x1x3)
|
||
|
|
*
|
||
|
|
* See also:
|
||
|
|
*
|
||
|
|
* subset
|
||
|
|
*
|
||
|
|
* @param {Matrix | Array} x Matrix to be squeezed
|
||
|
|
* @return {Matrix | Array} Squeezed matrix
|
||
|
|
*/
|
||
|
|
return typed(name, {
|
||
|
|
Array: function Array(x) {
|
||
|
|
return (0, _array.squeeze)((0, _object.clone)(x));
|
||
|
|
},
|
||
|
|
Matrix: function Matrix(x) {
|
||
|
|
var res = (0, _array.squeeze)(x.toArray()); // FIXME: return the same type of matrix as the input
|
||
|
|
|
||
|
|
return Array.isArray(res) ? matrix(res) : res;
|
||
|
|
},
|
||
|
|
any: function any(x) {
|
||
|
|
// scalar
|
||
|
|
return (0, _object.clone)(x);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
});
|
||
|
|
exports.createSqueeze = createSqueeze;
|