~/jendhub

def save_npy

stelliferous@0.0.2.0 · tensor_file.bend:420

import stelliferous@0.0.2.0/tensor_file.bend as TensorFile
import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/tensor_layout.bend as Layout

420# Writes a .npy file that NumPy loads with the tensor's shape.
421def save_npy(+path: String,tensor: Tensor()) -> IO(Tensor()):
422  Tensors.then(~Tensor(),~List<&2,U32>,~IO(Tensor()),Tensors.shape(~F32,tensor),tensor => shape => npy_saving(path,Layout.copy(~F32,tensor),shape))

Uses (10)

Used by (2)