Module torch

Functions

▷ def deserialize_torch_Parameters(header, frames) @dask_deserialize.register(torch.nn.Parameter)
@dask_deserialize.register(torch.nn.Parameter)
def deserialize_torch_Parameters(
header,
frames,
)
source link
▷ def deserialize_torch_Tensor(header, frames) @dask_deserialize.register(torch.Tensor)
@dask_deserialize.register(torch.Tensor)
def deserialize_torch_Tensor(
header,
frames,
)
source link
▷ def serialize_torch_Parameters(p) @dask_serialize.register(torch.nn.Parameter)
@dask_serialize.register(torch.nn.Parameter)
def serialize_torch_Parameters(p)
source link
▷ def serialize_torch_Tensor(t) @dask_serialize.register(torch.Tensor)
@dask_serialize.register(torch.Tensor)
def serialize_torch_Tensor(t)
source link