Tensorflow.js tf.tensor6d() Function

Tensorflow.js is an open-source library developed by Google for running machine learning models and deep learning neural networks in the browser or node environment.

The .tensor6d() function is used to create a new 6-dimensional tensor with the parameters namely value, shape, and datatype.

Syntax:  

tf.tensor6d(value, shape?, dataType?)

Parameters:  

• value: The value of the tensor which can be a nested array of numbers, or a flat array, or a TypedArray.
• shape: It takes the shape of the tensor. The tensor will infer its shape from the value if it is not provided. It is an optional parameter.
• datatype: It can be a ‘float32’ or ‘int32’ or ‘bool’ or ‘complex64’ or ‘string’ value. It is an optional parameter.

Return Value:  It returns the tensor of the same data type. The returned tensor will always be 6-dimensional.

Note: The 6d tensor functionality can also be achieved using the tf.tensor() function, but using tf.tensor6d() makes the code easily understandable and readable.

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs";

// Create the tensor
var example1 = tf.tensor6d([[[[
    [[1, 3], [2, 8]],
    [[3, 9], [4, 2]]
]]]]);

// Print the tensor
example1.print()

Output:

Tensor
    [[[[[[1, 3],
         [2, 8]],

        [[3, 9],
         [4, 2]]]]]]

// Import the tensorflow.js library
import * as tf from "@tensorflow/tfjs"

// Define the value of the tensor
var value = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12];

// Specify the shape of the tensor
var shape = [1, 2, 6, 1, 1, 1];

// Create the tensor
var example2 = tf.tensor6d(value, shape);

// Print the tensor
example2.print();

Output:

Tensor
    [[[ [ [[1 ],]],
        [ [[2 ],]],
        [ [[3 ],]],
        [ [[4 ],]],
        [ [[5 ],]],
        [ [[6 ],]]],

      [ [ [[7 ],]],
        [ [[8 ],]],
        [ [[9 ],]],
        [ [[10],]],
        [ [[11],]],
        [ [[12],]]]]]

// Import the tensorflow.js library
import * as tf from "@tensorflow/tfjs";

// Define the value of the tensor
var value = ["C", "C++", "Java", "Python",
            "PHP", "JS", "SQL", "React"];

// Specify the shape of the tensor
var shape = [1, 2, 4, 1, 1, 1];

// Create the tensor
var example3 = tf.tensor6d(value, shape);

// Print the tensor
example3.print();

Output:

Tensor
    [[[ [ [['C'     ],]],
        [ [['C++'   ],]],
        [ [['Java'  ],]],
        [ [['Python'],]]],

      [ [ [['PHP'   ],]],
        [ [['JS'    ],]],
        [ [['SQL'   ],]],
        [ [['React' ],]]]]]