这很好用:
import TensorFlow
var t = Tensor<Float>([[1, 0], [0, 1]])
但是下面给出了错误
import TensorFlow
var a = [[1, 0], [0, 1]]
var t = Tensor<Float>(a)
error: expression type 'Tensor<Float>' is ambiguous without more context
var t = Tensor<Float>(a)
^~~~~~~~~~~~~~~~
为什么会这样?还有如何从数组中快速创建张量。
最佳答案
您的第一个代码之所以有效,是因为它使用文字,而不是使用已声明的变量(其类型已确定)来初始化 Tensor<Float>
.编译器对文字进行特殊处理。
Tensor.init
的过载你打电话的是this .在您的情况下,它接受 ShapedArray<Float>
.
ShapedArray
符合协议(protocol)ExpressibleByArrayLiteral
,这意味着编译器可以将数组文字转换为 ShapeArray
含蓄地。然而,这仅限于文字,例如[[1, 0], [0, 1]]
,而不是引用数组类型变量的标识符,例如 a
. a
的类型被推断为 [[Int]]
由编译器,编译器不能隐式转换 [[Int]]
至 ShapedArray<Float>
.
虽然不那么重要,但事实是 Float
符合 ExpressibleByIntegerLiteral
(因此 1
和 0
可以转换为 float )也在允许您的代码编译方面发挥作用。
为什么var aa: Tensor<Float> = [[1.0, 0.0], [1.0, 0.0]]
的原因工作是因为Tensor
也符合 ExpressibleByArrayLiteral
.
关于swift - 从 swift 数组创建张量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59508668/