(
values: TensorLike6D,
shape?: [number, number, number, number, number, number],
dtype?: DataType)
| 46 | * @doc {heading: 'Tensors', subheading: 'Creation'} |
| 47 | */ |
| 48 | export function tensor6d( |
| 49 | values: TensorLike6D, |
| 50 | shape?: [number, number, number, number, number, number], |
| 51 | dtype?: DataType): Tensor6D { |
| 52 | assertNonNull(values); |
| 53 | if (shape != null && shape.length !== 6) { |
| 54 | throw new Error('tensor6d() requires shape to have six numbers'); |
| 55 | } |
| 56 | const inferredShape = inferShape(values, dtype); |
| 57 | if (inferredShape.length !== 6 && inferredShape.length !== 1) { |
| 58 | throw new Error( |
| 59 | 'tensor6d() requires values to be number[][][][][][] or ' + |
| 60 | 'flat/TypedArray'); |
| 61 | } |
| 62 | if (inferredShape.length === 1 && shape == null) { |
| 63 | throw new Error( |
| 64 | 'tensor6d() requires shape to be provided when `values` ' + |
| 65 | 'are a flat array'); |
| 66 | } |
| 67 | shape = shape || |
| 68 | inferredShape as [number, number, number, number, number, number]; |
| 69 | return makeTensor(values, shape, inferredShape, dtype) as Tensor6D; |
| 70 | } |
no test coverage detected
searching dependent graphs…