@ai-on-browser/data-analysis-models/lib/model/nns/layer/shape.d.ts

Data analysis model package without any dependencies

/**
 * Shape layer
 */
export default class ShapeLayer extends Layer {
    calc(x: any): Tensor<any>;
    _i: any;
    _size: any;
    grad(): Tensor<number> | Matrix<number>;
}
import Layer from './base.js';
import Tensor from '../../../util/tensor.js';
import Matrix from '../../../util/matrix.js';