tensor.io – Tensor IO Ops#

File operation#

  • Load from disk with the function load and its associated op LoadFromDisk

Details#

pytensor.tensor.io.load(path, dtype, shape, mmap_mode=None)[source]#

Load an array from an .npy file.

Parameters:
  • path – A Generic symbolic variable, that will contain a string

  • dtype (data-type) – The data type of the array to be read.

  • shape – The static shape information of the loaded array.

  • mmap_mode – How the file will be loaded. None means that the data will be copied into an array in memory, ‘c’ means that the file will be mapped into virtual memory, so only the parts that are needed will be actually read from disk and put into memory. Other modes supported by numpy.load (‘r’, ‘r+’, ‘w+’) cannot be supported by PyTensor.

Examples

>>> from pytensor import *
>>> path = Variable(Generic(), None)
>>> x = tensor.load(path, "int64", (None,))
>>> y = x * 2
>>> fn = function([path], y)
>>> fn("stored-array.npy")  
array([0, 2, 4, 6, 8], dtype=int64)