我测试 tf.tensor.as4D(); 时出错

Question

有人可以向我解释为什么会出现此错误以及我在使用 4D 张量时应该怎么做。

I try this code and always fail on 4D tensor

谢谢大家

import * as tf from '@tensorflow/tfjs';

const x = tf.tensor([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]);

const row = 5;
const column = 5;
const depth = 1;
const depth2 = 1;

// Convert from 1D to 2D tensor.
x.as2D(row, column).print();

// Convert from 1D to 3D tensor
x.as3D(row, column, depth).print();

// Convert from 1D to 4D tensor
x.as4D(row, column, depth, depth2).print();

错误

ERROR Error: Uncaught (in promise): Error: Size (6) must match the product of shape 5,5 Error: Size (6) must match the product of shape 5,5

Answer 1

从错误文本来看，传入的张量 x 有 6 个元素，而不是形状 ( 5, 5, 1, 1 ) 表示的 25 个元素，因此错误与转换本身，但事实上您没有足够的元素来填充 25 个框。 :)

我测试 tf.tensor.as4D(); 时出错

Error when i test tf.tensor.as4D();

tensorflow

tensorflow.js

错误