https://github.com/tensorly/tensorly
Revision 0967e0cdd01304c234121ad89babb2ffccd2f4d2 authored by JeanKossaifi on 24 October 2016, 11:10:54 UTC, committed by JeanKossaifi on 24 October 2016, 11:10:54 UTC
1 parent 33ad3be
Tip revision: 0967e0cdd01304c234121ad89babb2ffccd2f4d2 authored by JeanKossaifi on 24 October 2016, 11:10:54 UTC
Added doc
Added doc
Tip revision: 0967e0c
plot_tensor.py
# -*- coding: utf-8 -*-
"""
Basic tensor operations
=======================
Example on how to use :mod:`tensorly.base` to perform basic tensor operations.
"""
import matplotlib.pyplot as plt
from tensorly.base import unfold, fold
import numpy as np
###########################################################################
# A tensor is simply a numpy array
tensor = np.arange(24).reshape((3, 4, 2))
print('* original tensor:\n{}'.format(tensor))
###########################################################################
# Unfolding a tensor is easy
for mode in range(tensor.ndim):
print('* mode-{} unfolding:\n{}'.format(mode, unfold(tensor, mode)))
###########################################################################
# Re-folding the tensor is as easy:
for mode in range(tensor.ndim):
unfolded = unfold(tensor, mode)
print('* mode-{} unfolding:\n{}'.format(mode, fold(unfolded, mode, tensor.shape)))
Computing file changes ...