Lines Matching refs:reshape
27 tensor2 = tensor1.reshape(dim1);
29 tensor3 = tensor1.reshape(dim2);
31 tensor4 = tensor1.reshape(dim1).reshape(dim3);
59 tensor3 = tensor1.reshape(newDims1).contract(tensor2.reshape(newDims2), contract_along);
77 tensor2d.reshape(dim) = tensor;
81 tensor5d.reshape(dim).device(Eigen::DefaultDevice()) = tensor;
160 Tensor<float, 1, DataLayout> tensor6 = tensor4.reshape(DSizes<ptrdiff_t, 1>(7*7)).exp().slice(DSizes<ptrdiff_t, 1>(0), DSizes<ptrdiff_t, 1>(35));
450 matrix.slice(DSizes<ptrdiff_t, 2>(2, 0), DSizes<ptrdiff_t, 2>(1, 11)).reshape(newDims);