有人可以向我解释为什么我有这个错误,以及在使用4D张量时我该怎么办。
我尝试使用此代码并始终在4D张量上失败
谢谢大家。
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();
错误错误:未捕获(承诺):错误:大小(6)必须与形状5,5的产品匹配 错误:大小(6)必须与形状5,5
的乘积匹配
答案 0 :(得分:2)
从错误文本中,传入张量x
有6个元素,而不是形状(5,5,1,1)所示的25个元素,因此错误与每个转换无关但事实上你没有足够的元素来填充25个盒子。 :)