pytaco.tensor_cube

pytaco.tensor_cube(t1, out_format, dtype=None)

Cubes each input in the tensor.

Parameters

t1: tensor, array_like

input tensor or array_like object

out_format: format, mode_format, optional

• If a format is specified, the result tensor is stored in the format out_format.

• If a mode_format is specified, the result the result tensor has a with all of the dimensions stored in the mode_format passed in.

dtype: Datatype

The datatype of the output tensor.

Returns

cube: tensor

The element wise cube of the input tensor.

Examples

>>> import pytaco as pt
>>> pt.tensor_cube([1, -2, 7], out_format=pt.dense).toarray()
array([  1,  -8, 343], dtype=int64)