ExTorch.Tensor.Options (extorch v0.2.0)

Copy Markdown

The ExTorch.Tensor.Options struct defines the creation parameters of a tensor.

Summary

Types

t()

An ExTorch.Tensor.Options defines the creation parameters of a tensor.

Types

t()

@type t() :: %ExTorch.Tensor.Options{
  device: ExTorch.Device.device(),
  dtype: ExTorch.DType.dtype(),
  layout: ExTorch.Layout.layout(),
  memory_format: ExTorch.MemoryFormat.memory_format(),
  pin_memory: boolean(),
  requires_grad: boolean()
}

An ExTorch.Tensor.Options defines the creation parameters of a tensor.