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

时间:2018-04-07 21:47:35

标签: tensorflow tensorflow.js

有人可以向我解释为什么我有这个错误,以及在使用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

的乘积匹配

1 个答案:

答案 0 :(得分:2)

从错误文本中,传入张量x有6个元素,而不是形状(5,5,1,1)所示的25个元素,因此错误与每个转换无关但事实上你没有足够的元素来填充25个盒子。 :)