stgraph.compiler.val.pytorch package¶

Submodules¶

stgraph.compiler.val.pytorch.torch_val module¶

class stgraph.compiler.val.pytorch.torch_val.TorchVal(backend, tensor, val_type, id, fprog, reduce_dim)[source]¶

Bases: Val

property backend¶: Return backend system

property backend_key¶: Return key of backend system

property device¶: Return which device is on

property dtype¶: Return the DType

property layout¶: Return the layout

property requires_grad¶: Return whether requires gradient

property size¶: Return the size(i.e. shape)

sum()[source]¶

property val_type¶: Return ValueType

view(*args, **kargs)[source]¶

Module contents¶